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

Added collections import

Browse source
This commit is contained in:
Brandon Rozek 2019-02-10 23:59:29 -05:00
parent b2ab2ee132
commit be637664e7
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
rltorch/action_selector/EpsilonGreedySelector.py
View file

@ -1,5 +1,6 @@
from .ArgMaxSelector import ArgMaxSelector from .ArgMaxSelector import ArgMaxSelector
import numpy as np import numpy as np
import collections
class EpsilonGreedySelector(ArgMaxSelector): class EpsilonGreedySelector(ArgMaxSelector):
def __init__(self, model, action_size, device = None, epsilon = 0.1): def __init__(self, model, action_size, device = None, epsilon = 0.1):
super(EpsilonGreedySelector, self).__init__(model, action_size, device = device) super(EpsilonGreedySelector, self).__init__(model, action_size, device = device)