remove plot func

mprhode · Jan 31, 2022 · ce67e35 · ce67e35
1 parent eacd30a
commit ce67e35
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/train.py b/train.py
@@ -103,8 +103,6 @@ def train(agent_name="Blue", gamma=0.99, epsilon=1, lr=0.0001,
             env = ChallengeWrapper(agent_name="Blue",
                                    env=CybORG(PATH, 'sim', agents=agent_list[agent_counter % len(agent_list)]))
     except KeyboardInterrupt:
-        plot_learning_curve(steps_array, scores, eps_history, FIGURE_FILE)
-
         agent.save_models()