mirror of
https://github.com/jcreek/Sarpy.git
synced 2026-07-13 03:03:43 +00:00
Josh Creek
21 lines
577 B
Python
21 lines
577 B
Python
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
|