SUD-PROMISE / saved_models /performance.json
paulnhuu174's picture
Upload 32 files
d107fd9 verified
{
"test_metrics": {
"Logistic_Regression": {
"AUC": 0.8412681094315042,
"Accuracy": 0.7679502369668246,
"F1": 0.7698184168772404
},
"Random_Forest": {
"AUC": 0.8273200989420723,
"Accuracy": 0.7533767772511848,
"F1": 0.717628705148206
},
"XGBoost": {
"AUC": 0.7347429724175111,
"Accuracy": 0.7026658767772512,
"F1": 0.610477299185099
},
"MM_DNN": {
"AUC": 0.8507875496956492,
"Accuracy": 0.7731635071090047,
"F1": 0.7757540263543192
}
},
"training_info": {
"train_size": 67520,
"test_size": 16880,
"positive_ratio": 0.5
}
}