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 Binary files a/requirements.txt and /dev/null differ
training_models/train_models.py
View file @ ee6f1885
...@@ -78,8 +78,8 @@ if __name__ == "__main__": ...@@ -78,8 +78,8 @@ if __name__ == "__main__":
# Defining the models to train # Defining the models to train
# -------------------------------------------------------------------------------------------------------- # --------------------------------------------------------------------------------------------------------
# 1. No class weight # 1. No class weight
models_1 = {"DT" : DecisionTreeClassifier(), models_1 = {#"DT" : DecisionTreeClassifier(),
# "RF" : RandomForestClassifier(), "RF" : RandomForestClassifier(),
# "Bagging" : BaggingClassifier(), # "Bagging" : BaggingClassifier(),
# "AB" : AdaBoostClassifier(), # "AB" : AdaBoostClassifier(),
# "XGB": XGBClassifier(), # "XGB": XGBClassifier(),
...@@ -90,8 +90,8 @@ if __name__ == "__main__": ...@@ -90,8 +90,8 @@ if __name__ == "__main__":
} }
# 2. Class weight # 2. Class weight
models_2 = {"DT" : DecisionTreeClassifier(class_weight='balanced'), models_2 = {#"DT" : DecisionTreeClassifier(class_weight='balanced'),
# "RF" : RandomForestClassifier(class_weight='balanced'), "RF" : RandomForestClassifier(class_weight='balanced'),
# "Bagging" : BaggingClassifier(), # <- # "Bagging" : BaggingClassifier(), # <-
# "AB" : AdaBoostClassifier(), # <- # "AB" : AdaBoostClassifier(), # <-
# "XGB": XGBClassifier(), # <- # "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