-
Notifications
You must be signed in to change notification settings - Fork 0
/
default_hyperparams.json
89 lines (89 loc) · 1.56 KB
/
default_hyperparams.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
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
{
"LR": {
"C": [1.0],
"penalty": ["l2"],
"class_weight": [null],
"solver": ["lbfgs"],
"l1_ratio": [null],
"max_iter": [100]
},
"ELASTICNET": {
"alpha": [1],
"l1_ratio": [0]
},
"RF": {
"n_estimators": [100],
"max_depth": [null],
"class_weight": [null]
},
"DT": {
"max_depth": [null],
"max_leaf_nodes": [null],
"class_weight": [null],
"splitter": ["best"]
},
"MLP": {
"hidden_layer_sizes": [[100]],
"alpha": [0.0001],
"solver": ["adam"],
"activation": ["relu"]
},
"XGB": {
"max_depth": [6],
"learning_rate": [0.3],
"n_estimators": [100]
},
"PYGAM": {
"n_splines": [20],
"lam": [0.6]
},
"EBM": {
"max_bins": [256],
"interactions": [10],
"outer_bags": [8],
"inner_bags": [0]
},
"GAMINET": {
"interact_num": [20],
"activation_func": ["ReLU"],
"reg_clarity": [0.1]
},
"EXNN": {
"subnet_num": [5],
"l1_proj": [0.001],
"l1_subnet": [0.001]
},
"NAM": {
"lr": [0.02082],
"num_learners": [1],
"dropout": [0.1],
"num_basis_functions": [64]
},
"IGANN": {
"boost_rate": [0.1],
"elm_scale": [1],
"interactions": [0]
},
"TABNET": {
"n_a_and_d": [8],
"n_steps": [3],
"gamma": [1.3]
},
"CATBOOST": {
"n_estimators": [1000],
"eta": [0.03],
"max_depth": [6]
},
"MGCV_SPLINE": {
"spline_type": ["ts"],
"maxk": [10],
"m": [2],
"gamma": [1]
},
"GAMENS": {
"rsm_size": [2],
"num_classifiers": [10],
"df": [4],
"fusion": ["avgagg"]
}
}