def __init__(self):
#self.memory = MemoryD(self.memory_size)
self.memory = DataSet(80, 80, self.memory_size, 4)
self.ale = ALE(display_screen="true",
skip_frames=4,
game_ROM='../libraries/ale/roms/breakout.bin')
self.nnet = NeuralNet(self.state_size,
self.number_of_actions,
"ai/deepmind-layers.cfg",
"ai/deepmind-params.cfg",
"layer4",
discount_factor=self.discount_factor)
def __init__(self):
#self.memory = MemoryD(self.memory_size)
self.memory = DataSet(80, 80, self.memory_size, 4)
self.ale = ALE(display_screen="true",
skip_frames=4,
game_ROM='../libraries/ale/roms/breakout.bin')
#self.nnet = NeuralNet(self.state_size, self.number_of_actions, "ai/deepmind-layers.cfg", "ai/deepmind-params.cfg", "layer4")
self.nnet = CNNQLearner(self.number_of_actions,
4,
80,
80,
discount=.9,
learning_rate=.0001,
batch_size=32,
approximator='none')
def __init__(self):
# initialize memory
#self.memory = MemoryD(self.memory_size)
self.memory = DataSet(self.image_size, self.image_size,
self.memory_size, 4)
# initalize ALE
self.ale = ALE(display_screen="true",
skip_frames=4,
game_ROM='../libraries/ale/roms/breakout.bin',
preprocess_type=self.preprocess_type)
# initialize neural network
#self.nnet = NeuralNet(self.state_size, self.number_of_actions, "ai/deepmind-layers.cfg",
# "ai/deepmind-params.cfg", "layer4", discount_factor= self.discount_factor)
self.nnet = CNNQLearner(self.number_of_actions,
4,
self.image_size,
self.image_size,
discount=self.discount_factor,
learning_rate=.0001,
batch_size=32,
approximator='cuda_conv')
