ajoubrel-ensae/BDC-team-1
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

generalization #11

Merged
arevelle-ensae merged 10 commits from generalization into main 2024-03-28 09:40:04 +01:00
Conversation 0 Commits 10 Files Changed 4 +5
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit adc62dd056 - Show all commits

5
0_5_Machine_Learning.py
View File

@ -63,6 +63,7 @@ model_result = pd.DataFrame(columns= ["Model", "Accuracy", "Recall", "F1_score",
# Naive Bayes
model_result = pipeline_naiveBayes_benchmark(X_train, y_train, X_test, y_test, model_result)
save_result_set_s3(model_result , "resultat", type_of_activity, type_of_model)
print("Naive Bayes : Done")
# Logistic Regression
@ -70,12 +71,16 @@ model_result = pipeline_logreg_benchmark(X_train, y_train, X_test, y_test, model
print("Logistic : Done")
model_result = pipeline_logreg_cv(X_train, y_train, X_test, y_test, model_result)
save_result_set_s3(model_result , "resultat", type_of_activity, type_of_model)
print("Logistic CV : Done")
# Random Forest
model_result = pipeline_randomF_benchmark(X_train, y_train, X_test, y_test, model_result)
save_result_set_s3(model_result , "resultat", type_of_activity, type_of_model)
print("Random Forest : Done")
model_result = pipeline_randomF_cv(X_train, y_train, X_test, y_test, model_result)
save_result_set_s3(model_result , "resultat", type_of_activity, type_of_model)
print("Random Forest CV: Done")
# Save result