File size: 1,730 Bytes
2a33798
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
import numpy as np

# https://colab.research.google.com/drive/1DdWsGi10232orUv-reY4wsTmT0VMoHaX?usp=sharing#scrollTo=4OfVmDKk7XvG
# LLMs bias on 0 so make the actions 1, 2, 3, 4, 5 and 6 instead.

def dedicated_1_policy(state, pre_action=1):
    def get_description():   
        return "Always select action 1"
    dedicated_1_policy.description = get_description()
    return 1

def dedicated_2_policy(state, pre_action=1):
    def get_description():
        return "Always select action 2"
    dedicated_2_policy.description = get_description()
    return 2

def dedicated_3_policy(state, pre_action=1):
    def get_description():
        return "Always select action 3"
    dedicated_3_policy.description = get_description()
    return 3

def dedicated_4_policy(state, pre_action=1):
    def get_description():
        return "Always select action 4"
    dedicated_4_policy.description = get_description()
    return 4

def dedicated_5_policy(state, pre_action=1):
    def get_description():
        return "Always select action 5"
    dedicated_5_policy.description = get_description()
    return 5

def dedicated_6_policy(state, pre_action=1):
    def get_description():
        return "Always select action 6"
    dedicated_6_policy.description = get_description()
    return 6

def pseudo_random_policy(state, pre_action):
    def get_description():
        return "Select action from 1 to 6 alternatively"
    pseudo_random_policy.description = get_description()
    return pre_action % 6 + 1

def real_random_policy(state,pre_action=1):
    def get_description():
        return "Select action with a random policy"
    real_random_policy.description = get_description()
    return np.random.choice([1, 2, 3, 4, 5, 6])