Example #1
0
 def copy_batch(self):
     """
     Make a copy of the current batch
     :return: the copied batch
     """
     b2 = Batch()
     for i in range(self.size):
         ep = Episode()
         sep = self.episodes[i]
         for j in range(self.episodes[i].len):
             ep.add(sep.state_pool[j], sep.action_pool[j], sep.reward_pool[j],
                    sep.done_pool[j], sep.next_state_pool[j])
         b2.add_episode(ep)
     return b2
Example #2
0
 def _update_episodes(self):
     for entry in self._rss.entries:
         Episode.add(entry, self)