RPlagueInc/parameters.py at Final · Bender97/RPlagueInc · GitHub

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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119

#########################################
# NET PARAMS
#########################################

NET_GAMMA = 0.9
LEARNING_RATE = 0.001

EPOCHS = 20

MEMORY_SIZE = 20000
BATCH_SIZE = 32
STATUS_WINDOW = 6
N_STEPS = 10

EXPLORATION_MAX = 1.0
EXPLORATION_MIN = 0.01
EXPLORATION_DECAY = 0.995

YAML_PATH 	= "model.yaml"
MODEL_PATH 	= "model.h5"

#########################################
#########################################


#########################################
# SIMULATION PARAMETERS
#########################################

ENV_NAME = "engine-v0"

INITIAL_CASES 	= 5
N_HOUSES 		= 100
MEASURE_DURATION= 14

VIRUS_RANGE 		= 2
VIRUS_P_INFECTION 	= 0.3
VIRUS_SEVERITY 		= 0.8
VIRUS_LETHALITY 	= 0.4

INCUBATION_DURATION_RANGE = (2, 14)
DISEASE_DURATION_RANGE = (2, 28)
ASYMPTOMATIC_DURATION_RANGE = (7, 28)

# discontent values
FOOD_DAILY_DISCONTENT		= 100	# df
LEISURE_DAILY_DISCONTENT	= 10	# dl

MAX_MONEY_PER_HOUSE = 10000

# infection delta curve param
RELU_PARAM = 0.2

# reward weights
ALPHA = 10 # h(n) susc. + recov.
BETA = -100 # delta i(n) infected gradient
GAMMA = 600 # M(n) mean wealth
DELTA = -30 # D(n) discontent
EPSILON = -20 # d(n) deads
ZETA = -300 # delta d(n) deads gradient
ACT_PENALTY = 0 # choice taken in turn n

#########################################
#########################################


#########################################
# HEALTHSTATE STATUS PARAMETERS
#########################################

goodHealth_level = 0.8
badHealth_level = 0.5

old_age = 65
young_age = 20

SUSCEPTIBLE = 0
INCUBATION = 1
INFECTED = 2
ASYMPTOMATIC = 3
RECOVERED_FROM_ASYMPTOMATIC = 4
RECOVERED_FROM_INFECTED = 5
DEAD = 6

statusNum = 7

#########################################
#########################################


#########################################
# RENDERING PARAMETERS
#########################################

WHITE = ( 255, 255, 255 )

colors = (
	WHITE,				# susceptible
	( 50, 50, 255),		# incubation
	( 255, 50, 50),		# INFECTED
	( 255, 255, 50),	# asymptomatic
	( 50, 255, 50),		# recovered_from_asymptomatic
	( 50, 255, 50),		# recovered_from_infected
	)

FONTSIZE = 15

PADDING = 20
BORDER = 20

MONEYOFFSET = 17

FPS = 30

#########################################
#########################################