mirror of
https://github.com/jcreek/Sarpy.git
synced 2026-07-13 03:03:43 +00:00
feat(*): Add default bot code
This commit is contained in:
@@ -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
|
||||
Reference in New Issue
Block a user