/
pick_strategy.py
36 lines (27 loc) · 1.38 KB
/
pick_strategy.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
__author__ = 'anthony bell'
from pick_hole_agents.pick_agent import pick_agent
from pick_hole_agents.random_agent import random_agent
from pick_hole_agents.random_subset_agent import random_subset_agent
from pick_hole_agents.sequence_agent import sequence_agent
from pick_hole_agents.distribution_agent import distribution_agent
from pick_hole_agents.eta_greedy_RL_agent import eta_greedy_RL_agent
from pick_hole_agents.softmax_RL_agent import softmax_RL_agent
class pick_strategy():
@staticmethod
def random_agent(num_holes, name):
return random_agent(num_holes, name)
@staticmethod
def random_subset_agent(num_holes, holes_to_consider, name):
return random_subset_agent(num_holes, holes_to_consider, name)
@staticmethod
def eta_greedy_RL_agent(num_holes, name, init_Q, eta, step_size=None):
return eta_greedy_RL_agent(num_holes, name, init_Q, eta, step_size)
@staticmethod
def softmax_RL_agent(num_holes, name, init_Q, temp, step_size=None):
return softmax_RL_agent(num_holes, name, init_Q, temp, step_size)
@staticmethod
def sequence_agent(num_holes, holes_to_consider, time_to_consider, name):
return sequence_agent(num_holes, holes_to_consider, time_to_consider, name)
@staticmethod
def distribution_agent(num_holes, holes_distribution, name):
return distribution_agent(num_holes, holes_distribution, name)