save_agent.py

import os
from datetime import datetime as dt

import numpy as np


def save_results(agent, arr):
    # Save array of Q updates (or anything else) and accompanying agent metadata
    filename = '{}_{}'.format(dt.now().strftime('%H%M%S'), agent.algo_name)

    # Save
    np.save(os.path.join('.', 'saved_agents', filename), arr)
    f = open(os.path.join('.', 'saved_agents', '{}.txt'.format(filename)), 'w')
    for attr, value in player.__dict__.items():
        f.write(attr, value)


    return filename