Method push

intermediate_source/reinforcement_q_learning.py:143–145

Save a transition

(self, *args)

        self.memory = deque([], maxlen=capacity)

    def push(self, *args):
        """Save a transition"""
        self.memory.append(Transition(*args))

    def sample(self, batch_size):
        return random.sample(self.memory, batch_size)