os.mkdir("params")
except Exception:
pass
# handle loading / saving weights
savedPath = fname + "_final"
if os.path.exists("nnets/" + savedPath): # ugly, but as in dumpNetwork
print("Loading saved net: " + savedPath)
agent.setNetwork(savedPath)
if test:
agent.startMode(PLE_env.VALIDATION_MODE, 10000)
dump(vars(parameters), "params/" + fname + ".jldump")
agent.run(parameters.epochs, parameters.steps_per_epoch)
if test:
agent.summarizeTestPerformance()
else:
# -- save network
agent.dumpNetwork(savedPath)
# --- Show results ---
basename = "scores/" + fname
scores = load(basename + "_scores.jldump")
plt.plot(range(1, len(scores['vs'])+1), scores['vs'], label="VS", color='b')
plt.legend()
plt.xlabel("Number of epochs")
plt.ylabel("Score")
plt.savefig(basename + "_scores.pdf")
plt.show()
