Gym Environment¶

class olympus.reinforcement.gymenv.GymEnvironment(env_name, transforms=None, rand_seed=None, train_size=1024, valid_size=128, test_size=128, parallel_env=8, num_thread=4, distribution_mode='easy')[source]¶

Bases: object

Attributes:	action_space `input_size` Return the size of the samples state_space `target_size` Return the size of the target test train valid

Methods

categories() Dataset tags so we can filter what we want depending on the task

close
load_state_dict
max
sample_action
state_dict

action_space¶

static categories()[source]¶: Dataset tags so we can filter what we want depending on the task

close()[source]¶

input_size¶: Return the size of the samples

load_state_dict(data)[source]¶

max()[source]¶

sample_action()[source]¶

state_dict()[source]¶

state_space¶

target_size¶: Return the size of the target

test¶

train¶

valid¶