feat(*): Add default bot code

This commit is contained in:
Josh Creek
2023-09-25 20:37:02 +01:00
parent a5ac72216b
commit d12d42f114
11 changed files with 448 additions and 0 deletions
+20
View File
@@ -0,0 +1,20 @@
import os
class Agent:
def __init__(self):
# If you need to load your model from a file this is the time to do it
# You can do something like:
#
# self.actor = # your Model
#
# cur_dir = os.path.dirname(os.path.realpath(__file__))
# with open(os.path.join(cur_dir, 'model.p'), 'rb') as file:
# model = pickle.load(file)
# self.actor.load_state_dict(model)
pass
def act(self, state):
# Evaluate your model here
action = [1, 0, 0, 0, 0, 0, 0, 0]
return action