forked from eliteservice1002/mlmodels
-
Notifications
You must be signed in to change notification settings - Fork 0
/
gluon_automl.json
executable file
·54 lines (45 loc) · 1.24 KB
/
gluon_automl.json
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
{
"test": {
"data_pars": {
"train": true,
"uri_type": "csv",
"data_path": "../mlmodels/dataset/tabular/titanic_train_preprocessed.csv",
"col_target": "Survived"
},
"model_pars": {
"model_type": "tabular",
"learning_rate_min": 1e-4,
"learning_rate_max": 1e-2,
"learning_rate_default": 5e-4,
"activation": [
"relu",
"softrelu",
"tanh"
],
"layers": [
[100 ],
[1000 ],
[200, 100 ],
[300, 200, 100 ]
],
"dropout_prob_min": 0.0,
"dropout_prob_max": 0.5,
"dropout_prob_default": 0.1,
"num_boost_round": 100,
"num_leaves_lower": 26,
"num_leaves_upper": 66,
"num_leaves_default": 36
},
"compute_pars": {
"hp_tune": true,
"num_epochs": 10,
"time_limits": 60,
"num_trials": 5,
"search_strategy": "skopt"
},
"out_pars": {
"out_path": "dataset/"
}
},
"prod": {}
}