gfn.containers.replay_buffer
Module Contents
Classes
- class gfn.containers.replay_buffer.ReplayBuffer(env, loss_fn=None, objects_type=None, capacity=1000)
- Parameters
env (gfn.envs.Env) –
loss_fn (Loss | None) –
objects_type (Literal['transitions', 'trajectories', 'states'] | None) –
capacity (int) –
- __len__()
- __repr__()
Return repr(self).
- add(training_objects)
- Parameters
training_objects (Transitions | Trajectories | tuple[States]) –
- load(directory)
- Parameters
directory (str) –
- sample(n_trajectories)
- Parameters
n_trajectories (int) –
- Return type
Transitions | Trajectories | tuple[States]
- save(directory)
- Parameters
directory (str) –