get it working and log the time it took

chairbender · Jan 3, 2018 · d3a27a4 · d3a27a4
1 parent 06b926e
commit d3a27a4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -7,3 +7,10 @@ pip install .
 python learn.py
 ````
 
+# Performance
+
+This shows how long it takes to solve each task.
+
+| Task  | Agent | Episodes |
+| ------------------------------------------------------------------- | ------------------------------ | ----- |
+| FantasyFootballAuction-2OwnerSingleRosterSimpleScriptedOpponent-v0  | ShallowDQNFantasyFootballAgent | 23313 |
diff --git a/fantasy_football_auction_ai/agents.py b/fantasy_football_auction_ai/agents.py
@@ -103,7 +103,7 @@ def learn(self, train_steps=1000, test_episodes=100, plot=True):
             # check for it being solved
             if all(reward > .99 for reward in history['episode_reward']):
                 print("Solved in " + str(self.train_episodes) + " episodes of training.")
-                agent.save_weights('dqn_{}_params.h5f'.format(self.env.id), overwrite=True)
+                agent.save_weights('dqn_{}_params.h5f'.format(self.env.spec.id), overwrite=True)
                 plt.show()
                 break