Skip to content

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

trying RF

parent cb2c95a1
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
requirements.txt deleted 100644 → 0
View file @ cb2c95a1
Binary files a/requirements.txt and /dev/null differ
training_models/train_models.py
View file @ ee6f1885
......@@ -78,8 +78,8 @@ if __name__ == "__main__":
# Defining the models to train
# --------------------------------------------------------------------------------------------------------
# 1. No class weight
models_1 = {"DT" : DecisionTreeClassifier(),
# "RF" : RandomForestClassifier(),
models_1 = {#"DT" : DecisionTreeClassifier(),
"RF" : RandomForestClassifier(),
# "Bagging" : BaggingClassifier(),
# "AB" : AdaBoostClassifier(),
# "XGB": XGBClassifier(),
......@@ -90,8 +90,8 @@ if __name__ == "__main__":
}
# 2. Class weight
models_2 = {"DT" : DecisionTreeClassifier(class_weight='balanced'),
# "RF" : RandomForestClassifier(class_weight='balanced'),
models_2 = {#"DT" : DecisionTreeClassifier(class_weight='balanced'),
"RF" : RandomForestClassifier(class_weight='balanced'),
# "Bagging" : BaggingClassifier(), # <-
# "AB" : AdaBoostClassifier(), # <-
# "XGB": XGBClassifier(), # <-
......
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