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