{ | |
"svm": [ | |
{"C": [1, 10, 100], "gamma": [0.01, 0.001], "kernel": ["linear"]}, | |
{"C": [1, 10, 100], "gamma": [0.01, 0.001],"kernel": ["rbf"]} | |
], | |
"rf": { | |
"max_depth": [3], | |
"random_state": [0], | |
"min_samples_split": [2], | |
"min_samples_leaf": [1], | |
"n_estimators": [20], | |
"bootstrap": [true, false], | |
"criterion": ["gini", "entropy"] | |
}, | |
"pred_cols": ["Age", "Pclass", "Sex", "Fare"], | |
"dep_var": "Survived" | |
} |