-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.json
48 lines (48 loc) · 1.17 KB
/
config.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
{
"params": {
"cleaning": {
"filna_map": {
"Age": "mean"
},
"filter_cols": [
"PassengerId",
"Pclass",
"Sex",
"Age",
"SibSp",
"Parch",
"Fare",
"Embarked",
"Survived"
]
},
"feature_eng": {
"dummify_feature": [
"Pclass",
"Sex",
"Embarked"
]
},
"modelling": {
"name": "survival_classifier",
"save_path": "models/{}",
"target": "Survived",
"cv_folds": 10,
"numeric_features": [
"Age",
"Fare",
"SibSp",
"Parch"
],
"train_size": 0.75,
"normalize": 1,
"ignore_low_variance": 1,
"combine_rare_levels": 1,
"remove_multicollinearity": 1,
"feature_selection": 1,
"silent": 1,
"optimize_metric": "F1",
"optimize_iter": 15
}
}
}