cleanup and fix

README.md

@@ -34,9 +34,9 @@ More details about the project and the trained agent can be found in the [projec
 import gym
 import yaml
 import torch
-from helpers.qnn import QuantumNet
-from helpers.wrappers import BinaryWrapper
-from helpers.agent import Agent
+from model.qnn import QuantumNet
+from hemodellpers.wrappers import BinaryWrapper
+from model.agent import Agent
 
 # Environment
 env_name = 'FrozenLake-v1'

example.py

@@ -1,9 +1,9 @@
 import gym
 import yaml
 import torch
-from helpers.qnn import QuantumNet
-from helpers.wrappers import BinaryWrapper
-from helpers.agent import Agent
+from model.qnn import QuantumNet
+from model.wrappers import BinaryWrapper
+from model.agent import Agent
 
 # Environment
 env_name = 'FrozenLake-v1'

model/agent.py

@@ -1,5 +1,3 @@
-# Based on : https://github.com/djbyrne/core_rl/blob/master/algos/dqn/model.py
-
 import numpy as np
 import torch