Skip to content

C
covid_analysis
Commit 720e4ae1 authored by Joaquin Torres's avatar Joaquin Torres
Browse files
Options

Ready to re run everything

parent 1977e62f
Hide whitespace changes
Inline Side-by-side
Showing with 12 additions and 12 deletions
model_selection/hyperparam_tuning.py
View file @ 720e4ae1
......@@ -72,22 +72,22 @@ if __name__ == "__main__":
# --------------------------------------------------------------------------------------------------------
# 1. No class weight
models_simple = {"DT" : DecisionTreeClassifier(),
# "RF" : RandomForestClassifier(),
# "Bagging" : BaggingClassifier(),
# "AB" : AdaBoostClassifier(algorithm='SAMME'),
# "XGB": XGBClassifier(),
# "LR" : LogisticRegression(max_iter=1000),
# "SVM" : SVC(probability=True, max_iter=1000),
# "MLP" : MLPClassifier(max_iter=500)
"RF" : RandomForestClassifier(),
"Bagging" : BaggingClassifier(),
"AB" : AdaBoostClassifier(algorithm='SAMME'),
"XGB": XGBClassifier(),
"LR" : LogisticRegression(max_iter=1000),
"SVM" : SVC(probability=True, max_iter=1000),
"MLP" : MLPClassifier(max_iter=500)
}
# 2. Class weight: cost-sensitive learning
models_CS = {"DT" : DecisionTreeClassifier(class_weight='balanced'),
# "RF" : RandomForestClassifier(class_weight='balanced'),
# "Bagging" : BaggingClassifier(estimator= DecisionTreeClassifier(class_weight='balanced')),
# "AB" : AdaBoostClassifier(estimator= DecisionTreeClassifier(class_weight='balanced'), algorithm='SAMME'),
# "LR" : LogisticRegression(max_iter=1000, class_weight='balanced'),
# "SVM" : SVC(probability=True, max_iter = 1000, class_weight='balanced'),
"RF" : RandomForestClassifier(class_weight='balanced'),
"Bagging" : BaggingClassifier(estimator= DecisionTreeClassifier(class_weight='balanced')),
"AB" : AdaBoostClassifier(estimator= DecisionTreeClassifier(class_weight='balanced'), algorithm='SAMME'),
"LR" : LogisticRegression(max_iter=1000, class_weight='balanced'),
"SVM" : SVC(probability=True, max_iter = 1000, class_weight='balanced')
}
# --------------------------------------------------------------------------------------------------------
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment