Added collections import
This commit is contained in:
parent
b2ab2ee132
commit
be637664e7
1 changed files with 1 additions and 0 deletions
|
@ -1,5 +1,6 @@
|
|||
from .ArgMaxSelector import ArgMaxSelector
|
||||
import numpy as np
|
||||
import collections
|
||||
class EpsilonGreedySelector(ArgMaxSelector):
|
||||
def __init__(self, model, action_size, device = None, epsilon = 0.1):
|
||||
super(EpsilonGreedySelector, self).__init__(model, action_size, device = device)
|
||||
|
|
Loading…
Reference in a new issue