Simulation and Rendering

Simulate a policy in the given environment, optionally render a video.

class jax_agents.common.simulation.SimLogger(folder, field_names)

Bases: object

Log data from simulation in a csv file.

close()

Close file.

log(log_list)

Append data in file.

jax_agents.common.simulation.render_csv(env, folder, timesteps)

Load the csv file from the simulation and render it.

jax_agents.common.simulation.simulate(environment, algorithm, timesteps, folder)

Simulate one episode.