1
0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 11:10:20 +01:00

Unify base class using new-style (object).

This commit is contained in:
cxx
2017-10-26 10:16:11 +08:00
committed by Itai Caspi
parent 39cf78074c
commit f43c951c2d
16 changed files with 28 additions and 28 deletions

View File

@@ -19,7 +19,7 @@ import copy
from configurations import *
class Memory:
class Memory(object):
def __init__(self, tuning_parameters):
"""
:param tuning_parameters: A Preset class instance with all the running paramaters
@@ -43,7 +43,7 @@ class Memory:
pass
class Episode:
class Episode(object):
def __init__(self):
self.transitions = []
# a num_transitions x num_transitions table with the n step return in the n'th row
@@ -122,7 +122,7 @@ class Episode:
return batch
class Transition:
class Transition(object):
def __init__(self, state, action, reward, next_state, game_over):
self.state = copy.deepcopy(state)
self.state['observation'] = np.array(self.state['observation'], copy=False)