def main():
"""
This function will be called for training phase.
"""
# Sample code for illustration, add your code below to run in test phase.
# Load trained model from train/ directory
env = gym.make(MINERL_GYM_ENV)
if FRAME_SKIP > 0:
env = FrameSkip(env, enable_rendering=True)
env = ObsWrapper(env)
env = MoveAxisWrapper(env, -1, 0)
env = CombineActionWrapper(env)
agent = Agent(env.observation_space, env.action_space)
agent.load_model()
for _ in range(MINERL_MAX_EVALUATION_EPISODES):
obs = env.reset()
done = False
netr = 0
while not done:
action = agent.act(obs)
obs, reward, done, info = env.step(action)
netr += reward
env.render()
env.close()
