Fixed scoping error with Transitions
This commit is contained in:
parent
ad75539776
commit
07c90a09f9
1 changed files with 5 additions and 1 deletions
|
@ -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):
|
||||||
|
|
Loading…
Reference in a new issue