Text-Gym-Agents / envs /mujoco /walker2d_policies.py
CharlesZhang's picture
add 5 more envs: halfcheetah, hopper, walker2d, invertedPendulum, invertedDoublePendulum.
2ec5014
raw
history blame contribute delete
493 Bytes
import numpy as np
import random
def pseudo_random_policy(state, pre_action):
def get_description():
return "Select action randomly"
pseudo_random_policy.description = get_description()
return [2 * random.random() - 1 for i in range(6)]
def real_random_policy(state, pre_action=1):
def get_description():
return "Select action with a random policy"
real_random_policy.description = get_description()
return [2 * random.random() - 1 for i in range(6)]