brozek/rltorch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fixed scoping error with Transitions

Browse source
This commit is contained in:
Brandon Rozek 2019-11-04 12:09:09 -05:00
parent ad75539776
commit 07c90a09f9
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
rltorch/memory/DQfDMemory.py
View file

@ -1,4 +1,8 @@
from .PrioritizedReplayMemory import PrioritizedReplayMemory, Transition from .PrioritizedReplayMemory import PrioritizedReplayMemory
Transition = namedtuple('Transition',
('state', 'action', 'reward', 'next_state', 'done'))
class DQfDMemory(PrioritizedReplayMemory): class DQfDMemory(PrioritizedReplayMemory):
def __init__(self, capacity, alpha): def __init__(self, capacity, alpha):