rl_equation_solver.utilities.utilities.ReplayMemory

class ReplayMemory(capacity)[source]

Bases: object

Stores the Experience Replay buffer

Methods

push(*args)

Save the Experience into memory

sample(batch_size)

select a random batch of Experience for training

push(*args)[source]

Save the Experience into memory

sample(batch_size)[source]

select a random batch of Experience for training