ABM-Graph-Diffusion-Network/config.py at main · fracozzi/ABM-Graph-Diffusion-Network · 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
"""
Here set the configuration to train the model:

abm_model = 'schelling' or 'predator_prey'
model_type = 'surrogate' or 'ablation'

configuration = {'abm_model' : 'schelling',
                 'model_type':'surrogate',
                 'learning_rate':1e-5,
                 'T_diffusion':100,
                 'n_epochs':100
                }

OR

configuration = {'abm_model' : 'predator_prey',
                 'model_type':'ablation',
                 'learning_rate':1e-5,
                 'T_diffusion':100,
                 'n_epochs':100
                }

"""

configuration = {'abm_model' : 'predator_prey',
                 'model_type':'surrogate',
                 'learning_rate':1e-5,
                 'T_diffusion':100,
                 'n_epochs':100
                }