offlax.utils#

Code for utility functions used in Offlax

Functions

generate_offlax_dataset(env, agent, steps, path)

Generates an offlax dataset for the environment and agent
offlax.utils.generate_offlax_dataset(env, agent, steps, path)[source]#

Generates an offlax dataset for the environment and agent

Parameters

  • env (gym.Env) – An Gym API compatible environment

  • agent (Callable) – a function that returns an action given the state of the environment

  • steps (int) – number of steps

  • path (str) –

Return type

None