blob: 04a0890073a80c7d6f685075d360a5b5c777b49e [file] [log] [blame]
{
"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"
}