def navigation_main(): env = UnityEnvironment(file_name="Navigation/Banana.app") seed = 777 np.random.seed(seed) seed_torch(seed) num_episode = 2000 memory_size = 10000 batch_size = 64 target_update = 4 epsilon_decay = 0.9 agent = DQNAgent(env, memory_size, batch_size, target_update, epsilon_decay) agent.train(num_episode) agent.test()
# Import internal modules from agents.dqn_agent import DQNAgent # Import external modules import gym if __name__ == "__main__": # initialize gym environment and the agent env = gym.make('CartPole-v0') agent = DQNAgent(environment=env) # Test the agent agent.test()