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

Added missing dependency

Browse source
This commit is contained in:
Brandon Rozek 2019-02-03 21:55:08 -05:00
parent 0a6f1e73f3
commit f1a52cf77b
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
rltorch/action_selector/EpsilonGreedySelector.py
View file

@ -1,4 +1,5 @@
from .ArgMaxSelector import ArgMaxSelector
import numpy as np
class EpsilonGreedySelector(ArgMaxSelector):
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)