From 8815773a3bab6697a9a872cd36f48f812a4dac86 Mon Sep 17 00:00:00 2001
From: Collin Farquhar <farquhar13@gmail.com>
Date: Fri, 28 Jun 2024 12:17:53 -0600
Subject: [PATCH] remove example file

---
 training_models.py | 25 -------------------------
 1 file changed, 25 deletions(-)
 delete mode 100644 training_models.py

diff --git a/training_models.py b/training_models.py
deleted file mode 100644
index abfa05ef..00000000
--- a/training_models.py
+++ /dev/null
@@ -1,25 +0,0 @@
-from api.training import Training
-from models.agents import DQNAgent
-import gym
-from gym import spaces
-import numpy as np
-
-# Get Environment
-env = gym.make('CartPole-v1')
-
-class CartPoleWrapper(gym.ObservationWrapper):
-    def __init__(self, env):
-        super(CartPoleWrapper, self).__init__(env)
-        low = self.observation_space.low[2:]
-        high = self.observation_space.high[2:]
-        self.observation_space = spaces.Box(low=low, high=high, dtype=np.float32)
-
-    def observation(self, observation):
-        return observation[2:]
-
-env = CartPoleWrapper(env)
-
-agent = DQNAgent(2,2)
-train_api = Training(500, env, 32)
-train_api.train_model(agent, target_model=False)
-train_api.plot_performance('DQNAgent')