tanu/ML_AI_training
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

trying Stratified Kfold split on running multiple pipelines

Browse source
This commit is contained in:
Tanushree Tunstall 2022-03-09 18:35:54 +00:00
parent bb8f6f70ba
commit 1bfb35c30c
7 changed files with 287 additions and 72 deletions
Download patch file Download diff file Expand all files Collapse all files

12
pnca_results_v1.py
View file

@ -85,3 +85,15 @@ all_features: numerical_features + ['ss_class', 'wt_prop_water', 'mut_prop_water
9All XGBoost 0.710526 0.692308 0.729730 0.685714 0.683047)
#%%
Model F1_Score Precision Recall Accuracy ROC_AUC
0 Logistic Regression 0.757764 0.701149 0.824324 0.721429 0.715192
1 Naive Bayes 0.628571 0.666667 0.594595 0.628571 0.630631
2 K-Nearest Neighbors 0.666667 0.623529 0.716216 0.621429 0.615684
3 SVM 0.766467 0.688172 0.864865 0.721429 0.712735
4 MLP 0.726115 0.686747 0.770270 0.692857 0.688165
5 Decision Tree 0.647482 0.692308 0.608108 0.650000 0.652539
6 Extra Trees 0.760736 0.696629 0.837838 0.721429 0.714373
7 Random Forest 0.736196 0.674157 0.810811 0.692857 0.685708
8 Random Forest2 0.736196 0.674157 0.810811 0.692857 0.685708
9 XGBoost 0.710526 0.692308 0.729730 0.685714 0.683047)