33 lines
779 B
Python
33 lines
779 B
Python
|
from survival.components.time_component import TimeComponent
|
||
|
|
||
|
|
||
|
class LearningComponent:
|
||
|
def __init__(self):
|
||
|
self.made_step = False
|
||
|
self.old_state = None
|
||
|
self.action = None
|
||
|
self.resource = None
|
||
|
|
||
|
self.reward = 0
|
||
|
self.done = False
|
||
|
self.score = 0
|
||
|
self.record = 0
|
||
|
|
||
|
def load_step(self, old_state, action, resource):
|
||
|
self.old_state = old_state
|
||
|
self.action = action
|
||
|
if resource is None:
|
||
|
self.resource = None
|
||
|
else:
|
||
|
self.resource = resource
|
||
|
self.made_step = True
|
||
|
|
||
|
def reset(self):
|
||
|
self.made_step = False
|
||
|
self.old_state = None
|
||
|
self.action = None
|
||
|
self.resource = None
|
||
|
|
||
|
self.reward = 0
|
||
|
self.done = False
|