Veil commited on
Commit ·
b5b0d55
1
Parent(s): dfd7a82
Revert to state from commit 4ed03954
Browse files- README.md +15 -15
- calibration_curve.png +2 -2
- confusion_matrix.png +2 -2
- cv_fold_details.json +90 -90
- feature_importance.csv +100 -107
- feature_importance.png +2 -2
- gradient_boosting_metrics.json +18 -18
- gradient_boosting_model.joblib +2 -2
- learning_curves.png +2 -2
- metrics_comparison_across_folds.png +2 -2
- precision_recall_curve.png +2 -2
- roc_curve.png +2 -2
- training_summary.json +32 -32
- validation_curves.png +2 -2
README.md
CHANGED
|
@@ -18,7 +18,7 @@ This hospital-readmission-phase2-lgbm model predicts the risk of 30-day hospital
|
|
| 18 |
|
| 19 |
**Task:** Hospital 30-Day Readmission Risk Prediction
|
| 20 |
**Model Type:** Gradient Boosting Machine (LightGBM)
|
| 21 |
-
**Training Date:** 2025-12-
|
| 22 |
**Environment:** kaggle (GPU)
|
| 23 |
|
| 24 |
## Performance Metrics
|
|
@@ -27,28 +27,28 @@ This hospital-readmission-phase2-lgbm model predicts the risk of 30-day hospital
|
|
| 27 |
|
| 28 |
| Metric | Value |
|
| 29 |
|--------|-------|
|
| 30 |
-
| Mean ROC-AUC | 0.
|
| 31 |
|
| 32 |
### Final Test Set Results
|
| 33 |
|
| 34 |
#### Primary Metrics
|
| 35 |
| Metric | Value |
|
| 36 |
|--------|-------|
|
| 37 |
-
| ROC-AUC | 0.
|
| 38 |
-
| PR-AUC | 0.
|
| 39 |
-
| F1 Score | 0.
|
| 40 |
|
| 41 |
#### Classification Metrics
|
| 42 |
| Metric | Value |
|
| 43 |
|--------|-------|
|
| 44 |
-
| Precision | 0.
|
| 45 |
-
| Recall | 0.
|
| 46 |
|
| 47 |
#### Clinical Metrics
|
| 48 |
| Metric | Value |
|
| 49 |
|--------|-------|
|
| 50 |
-
| Sensitivity (TPR) | 0.
|
| 51 |
-
| Specificity (TNR) |
|
| 52 |
|
| 53 |
## Model Visualizations
|
| 54 |
|
|
@@ -81,7 +81,7 @@ This hospital-readmission-phase2-lgbm model predicts the risk of 30-day hospital
|
|
| 81 |
| Property | Value |
|
| 82 |
|----------|-------|
|
| 83 |
| Total Samples | 101,766 |
|
| 84 |
-
| Features |
|
| 85 |
| Development Set | 86,501 |
|
| 86 |
| Final Test Set | 15,265 |
|
| 87 |
|
|
@@ -98,20 +98,20 @@ This hospital-readmission-phase2-lgbm model predicts the risk of 30-day hospital
|
|
| 98 |
```python
|
| 99 |
{
|
| 100 |
"n_estimators": 150,
|
| 101 |
-
"learning_rate": 0.
|
| 102 |
-
"num_leaves":
|
| 103 |
"max_depth": -1,
|
| 104 |
"subsample": 0.7,
|
| 105 |
"colsample_bytree": 0.7,
|
| 106 |
-
"reg_alpha": 0.
|
| 107 |
"reg_lambda": 0.1
|
| 108 |
}
|
| 109 |
```
|
| 110 |
|
| 111 |
## Training Details
|
| 112 |
|
| 113 |
-
- **Total Training Time:**
|
| 114 |
-
- **Hyperparameter Search Time:**
|
| 115 |
- **Cross-Validation Folds:** 5
|
| 116 |
- **Early Stopping:** Yes
|
| 117 |
- **Device:** GPU
|
|
|
|
| 18 |
|
| 19 |
**Task:** Hospital 30-Day Readmission Risk Prediction
|
| 20 |
**Model Type:** Gradient Boosting Machine (LightGBM)
|
| 21 |
+
**Training Date:** 2025-12-15 10:55:05
|
| 22 |
**Environment:** kaggle (GPU)
|
| 23 |
|
| 24 |
## Performance Metrics
|
|
|
|
| 27 |
|
| 28 |
| Metric | Value |
|
| 29 |
|--------|-------|
|
| 30 |
+
| Mean ROC-AUC | 0.8399 ± 0.0055 |
|
| 31 |
|
| 32 |
### Final Test Set Results
|
| 33 |
|
| 34 |
#### Primary Metrics
|
| 35 |
| Metric | Value |
|
| 36 |
|--------|-------|
|
| 37 |
+
| ROC-AUC | 0.8424 |
|
| 38 |
+
| PR-AUC | 0.4000 |
|
| 39 |
+
| F1 Score | 0.1053 |
|
| 40 |
|
| 41 |
#### Classification Metrics
|
| 42 |
| Metric | Value |
|
| 43 |
|--------|-------|
|
| 44 |
+
| Precision | 0.6978 |
|
| 45 |
+
| Recall | 0.0569 |
|
| 46 |
|
| 47 |
#### Clinical Metrics
|
| 48 |
| Metric | Value |
|
| 49 |
|--------|-------|
|
| 50 |
+
| Sensitivity (TPR) | 0.0569 |
|
| 51 |
+
| Specificity (TNR) | 0.9969 |
|
| 52 |
|
| 53 |
## Model Visualizations
|
| 54 |
|
|
|
|
| 81 |
| Property | Value |
|
| 82 |
|----------|-------|
|
| 83 |
| Total Samples | 101,766 |
|
| 84 |
+
| Features | 113 |
|
| 85 |
| Development Set | 86,501 |
|
| 86 |
| Final Test Set | 15,265 |
|
| 87 |
|
|
|
|
| 98 |
```python
|
| 99 |
{
|
| 100 |
"n_estimators": 150,
|
| 101 |
+
"learning_rate": 0.05,
|
| 102 |
+
"num_leaves": 31,
|
| 103 |
"max_depth": -1,
|
| 104 |
"subsample": 0.7,
|
| 105 |
"colsample_bytree": 0.7,
|
| 106 |
+
"reg_alpha": 0.0,
|
| 107 |
"reg_lambda": 0.1
|
| 108 |
}
|
| 109 |
```
|
| 110 |
|
| 111 |
## Training Details
|
| 112 |
|
| 113 |
+
- **Total Training Time:** 214.37 minutes
|
| 114 |
+
- **Hyperparameter Search Time:** 128.14 minutes
|
| 115 |
- **Cross-Validation Folds:** 5
|
| 116 |
- **Early Stopping:** Yes
|
| 117 |
- **Device:** GPU
|
calibration_curve.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
confusion_matrix.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
cv_fold_details.json
CHANGED
|
@@ -2,24 +2,24 @@
|
|
| 2 |
{
|
| 3 |
"fold": 1,
|
| 4 |
"metrics": {
|
| 5 |
-
"roc_auc": 0.
|
| 6 |
-
"pr_auc": 0.
|
| 7 |
-
"precision": 0.
|
| 8 |
-
"recall": 0.
|
| 9 |
-
"f1": 0.
|
| 10 |
-
"accuracy": 0.
|
| 11 |
-
"balanced_accuracy": 0.
|
| 12 |
-
"sensitivity": 0.
|
| 13 |
-
"specificity": 0.
|
| 14 |
-
"ppv": 0.
|
| 15 |
-
"npv": 0.
|
| 16 |
-
"fpr":
|
| 17 |
-
"fnr": 0.
|
| 18 |
-
"true_positives":
|
| 19 |
-
"true_negatives":
|
| 20 |
-
"false_positives":
|
| 21 |
-
"false_negatives":
|
| 22 |
-
"brier_score": 0.
|
| 23 |
},
|
| 24 |
"train_size": 62280,
|
| 25 |
"val_size": 6920,
|
|
@@ -28,24 +28,24 @@
|
|
| 28 |
{
|
| 29 |
"fold": 2,
|
| 30 |
"metrics": {
|
| 31 |
-
"roc_auc": 0.
|
| 32 |
-
"pr_auc": 0.
|
| 33 |
-
"precision": 0.
|
| 34 |
-
"recall": 0.
|
| 35 |
-
"f1": 0.
|
| 36 |
-
"accuracy": 0.
|
| 37 |
-
"balanced_accuracy": 0.
|
| 38 |
-
"sensitivity": 0.
|
| 39 |
-
"specificity":
|
| 40 |
-
"ppv": 0.
|
| 41 |
-
"npv": 0.
|
| 42 |
-
"fpr": 0.
|
| 43 |
-
"fnr":
|
| 44 |
-
"true_positives":
|
| 45 |
-
"true_negatives":
|
| 46 |
-
"false_positives":
|
| 47 |
-
"false_negatives":
|
| 48 |
-
"brier_score": 0.
|
| 49 |
},
|
| 50 |
"train_size": 62280,
|
| 51 |
"val_size": 6921,
|
|
@@ -54,24 +54,24 @@
|
|
| 54 |
{
|
| 55 |
"fold": 3,
|
| 56 |
"metrics": {
|
| 57 |
-
"roc_auc": 0.
|
| 58 |
-
"pr_auc": 0.
|
| 59 |
-
"precision": 0.
|
| 60 |
-
"recall": 0.
|
| 61 |
-
"f1": 0.
|
| 62 |
-
"accuracy": 0.
|
| 63 |
-
"balanced_accuracy": 0.
|
| 64 |
-
"sensitivity": 0.
|
| 65 |
-
"specificity":
|
| 66 |
-
"ppv": 0.
|
| 67 |
-
"npv": 0.
|
| 68 |
-
"fpr": 0.
|
| 69 |
-
"fnr":
|
| 70 |
-
"true_positives":
|
| 71 |
-
"true_negatives":
|
| 72 |
-
"false_positives":
|
| 73 |
-
"false_negatives":
|
| 74 |
-
"brier_score": 0.
|
| 75 |
},
|
| 76 |
"train_size": 62280,
|
| 77 |
"val_size": 6921,
|
|
@@ -80,24 +80,24 @@
|
|
| 80 |
{
|
| 81 |
"fold": 4,
|
| 82 |
"metrics": {
|
| 83 |
-
"roc_auc": 0.
|
| 84 |
-
"pr_auc": 0.
|
| 85 |
-
"precision": 0.
|
| 86 |
-
"recall": 0.
|
| 87 |
-
"f1": 0.
|
| 88 |
-
"accuracy": 0.
|
| 89 |
-
"balanced_accuracy": 0.
|
| 90 |
-
"sensitivity": 0.
|
| 91 |
-
"specificity":
|
| 92 |
-
"ppv": 0.
|
| 93 |
-
"npv": 0.
|
| 94 |
-
"fpr": 0.
|
| 95 |
-
"fnr":
|
| 96 |
-
"true_positives":
|
| 97 |
-
"true_negatives":
|
| 98 |
-
"false_positives":
|
| 99 |
-
"false_negatives":
|
| 100 |
-
"brier_score": 0.
|
| 101 |
},
|
| 102 |
"train_size": 62280,
|
| 103 |
"val_size": 6921,
|
|
@@ -106,24 +106,24 @@
|
|
| 106 |
{
|
| 107 |
"fold": 5,
|
| 108 |
"metrics": {
|
| 109 |
-
"roc_auc": 0.
|
| 110 |
-
"pr_auc": 0.
|
| 111 |
-
"precision": 0.
|
| 112 |
-
"recall": 0.
|
| 113 |
-
"f1": 0.
|
| 114 |
-
"accuracy": 0.
|
| 115 |
-
"balanced_accuracy": 0.
|
| 116 |
-
"sensitivity": 0.
|
| 117 |
-
"specificity":
|
| 118 |
-
"ppv": 0.
|
| 119 |
-
"npv": 0.
|
| 120 |
-
"fpr": 0.
|
| 121 |
-
"fnr":
|
| 122 |
-
"true_positives":
|
| 123 |
-
"true_negatives":
|
| 124 |
-
"false_positives":
|
| 125 |
-
"false_negatives":
|
| 126 |
-
"brier_score": 0.
|
| 127 |
},
|
| 128 |
"train_size": 62280,
|
| 129 |
"val_size": 6921,
|
|
|
|
| 2 |
{
|
| 3 |
"fold": 1,
|
| 4 |
"metrics": {
|
| 5 |
+
"roc_auc": 0.8422628166877643,
|
| 6 |
+
"pr_auc": 0.383648852076946,
|
| 7 |
+
"precision": 0.5921052631578947,
|
| 8 |
+
"recall": 0.04660797514241326,
|
| 9 |
+
"f1": 0.08641382621219396,
|
| 10 |
+
"accuracy": 0.8900063580139876,
|
| 11 |
+
"balanced_accuracy": 0.5212870715009399,
|
| 12 |
+
"sensitivity": 0.04660797514241326,
|
| 13 |
+
"specificity": 0.9959661678594665,
|
| 14 |
+
"ppv": 0.5921052631578947,
|
| 15 |
+
"npv": 0.8926468015627733,
|
| 16 |
+
"fpr": 0.004033832140533507,
|
| 17 |
+
"fnr": 0.9533920248575868,
|
| 18 |
+
"true_positives": 90,
|
| 19 |
+
"true_negatives": 15308,
|
| 20 |
+
"false_positives": 62,
|
| 21 |
+
"false_negatives": 1841,
|
| 22 |
+
"brier_score": 0.08057624482185294
|
| 23 |
},
|
| 24 |
"train_size": 62280,
|
| 25 |
"val_size": 6920,
|
|
|
|
| 28 |
{
|
| 29 |
"fold": 2,
|
| 30 |
"metrics": {
|
| 31 |
+
"roc_auc": 0.8423694634254875,
|
| 32 |
+
"pr_auc": 0.39328310437895053,
|
| 33 |
+
"precision": 0.680327868852459,
|
| 34 |
+
"recall": 0.043005181347150256,
|
| 35 |
+
"f1": 0.08089668615984405,
|
| 36 |
+
"accuracy": 0.8909826589595375,
|
| 37 |
+
"balanced_accuracy": 0.5202338854035686,
|
| 38 |
+
"sensitivity": 0.043005181347150256,
|
| 39 |
+
"specificity": 0.997462589459987,
|
| 40 |
+
"ppv": 0.680327868852459,
|
| 41 |
+
"npv": 0.8924787518919548,
|
| 42 |
+
"fpr": 0.0025374105400130125,
|
| 43 |
+
"fnr": 0.9569948186528497,
|
| 44 |
+
"true_positives": 83,
|
| 45 |
+
"true_negatives": 15331,
|
| 46 |
+
"false_positives": 39,
|
| 47 |
+
"false_negatives": 1847,
|
| 48 |
+
"brier_score": 0.08036789416841797
|
| 49 |
},
|
| 50 |
"train_size": 62280,
|
| 51 |
"val_size": 6921,
|
|
|
|
| 54 |
{
|
| 55 |
"fold": 3,
|
| 56 |
"metrics": {
|
| 57 |
+
"roc_auc": 0.8372258386399722,
|
| 58 |
+
"pr_auc": 0.3786832966547781,
|
| 59 |
+
"precision": 0.5703703703703704,
|
| 60 |
+
"recall": 0.03989637305699482,
|
| 61 |
+
"f1": 0.07457627118644068,
|
| 62 |
+
"accuracy": 0.8895375722543353,
|
| 63 |
+
"balanced_accuracy": 0.5180613940756672,
|
| 64 |
+
"sensitivity": 0.03989637305699482,
|
| 65 |
+
"specificity": 0.9962264150943396,
|
| 66 |
+
"ppv": 0.5703703703703704,
|
| 67 |
+
"npv": 0.8920477716283134,
|
| 68 |
+
"fpr": 0.0037735849056603774,
|
| 69 |
+
"fnr": 0.9601036269430052,
|
| 70 |
+
"true_positives": 77,
|
| 71 |
+
"true_negatives": 15312,
|
| 72 |
+
"false_positives": 58,
|
| 73 |
+
"false_negatives": 1853,
|
| 74 |
+
"brier_score": 0.08114327062149242
|
| 75 |
},
|
| 76 |
"train_size": 62280,
|
| 77 |
"val_size": 6921,
|
|
|
|
| 80 |
{
|
| 81 |
"fold": 4,
|
| 82 |
"metrics": {
|
| 83 |
+
"roc_auc": 0.8318016530952919,
|
| 84 |
+
"pr_auc": 0.3557057624154197,
|
| 85 |
+
"precision": 0.51875,
|
| 86 |
+
"recall": 0.04298291040911445,
|
| 87 |
+
"f1": 0.07938785270205644,
|
| 88 |
+
"accuracy": 0.888728323699422,
|
| 89 |
+
"balanced_accuracy": 0.5189864125862997,
|
| 90 |
+
"sensitivity": 0.04298291040911445,
|
| 91 |
+
"specificity": 0.994989914763485,
|
| 92 |
+
"ppv": 0.51875,
|
| 93 |
+
"npv": 0.8921820303383897,
|
| 94 |
+
"fpr": 0.005010085236515063,
|
| 95 |
+
"fnr": 0.9570170895908856,
|
| 96 |
+
"true_positives": 83,
|
| 97 |
+
"true_negatives": 15292,
|
| 98 |
+
"false_positives": 77,
|
| 99 |
+
"false_negatives": 1848,
|
| 100 |
+
"brier_score": 0.08243304148954153
|
| 101 |
},
|
| 102 |
"train_size": 62280,
|
| 103 |
"val_size": 6921,
|
|
|
|
| 106 |
{
|
| 107 |
"fold": 5,
|
| 108 |
"metrics": {
|
| 109 |
+
"roc_auc": 0.8460232669561988,
|
| 110 |
+
"pr_auc": 0.3943230038407428,
|
| 111 |
+
"precision": 0.6230769230769231,
|
| 112 |
+
"recall": 0.04194717762817193,
|
| 113 |
+
"f1": 0.07860262008733625,
|
| 114 |
+
"accuracy": 0.8902312138728323,
|
| 115 |
+
"balanced_accuracy": 0.5193794707842857,
|
| 116 |
+
"sensitivity": 0.04194717762817193,
|
| 117 |
+
"specificity": 0.9968117639403995,
|
| 118 |
+
"ppv": 0.6230769230769231,
|
| 119 |
+
"npv": 0.8922539312754805,
|
| 120 |
+
"fpr": 0.0031882360596004944,
|
| 121 |
+
"fnr": 0.9580528223718281,
|
| 122 |
+
"true_positives": 81,
|
| 123 |
+
"true_negatives": 15320,
|
| 124 |
+
"false_positives": 49,
|
| 125 |
+
"false_negatives": 1850,
|
| 126 |
+
"brier_score": 0.08000282103315677
|
| 127 |
},
|
| 128 |
"train_size": 62280,
|
| 129 |
"val_size": 6921,
|
feature_importance.csv
CHANGED
|
@@ -1,121 +1,114 @@
|
|
| 1 |
feature,importance
|
| 2 |
-
discharge_disposition_id,
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
|
| 10 |
-
|
| 11 |
-
|
| 12 |
-
|
| 13 |
-
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
|
| 31 |
-
|
| 32 |
-
|
| 33 |
-
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
|
| 46 |
-
|
| 47 |
-
|
| 48 |
-
|
| 49 |
-
|
| 50 |
-
|
| 51 |
-
|
| 52 |
-
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 56 |
glipizide_Up,3
|
| 57 |
-
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
|
| 62 |
-
|
| 63 |
-
|
| 64 |
-
|
| 65 |
-
|
| 66 |
-
|
| 67 |
-
max_glu_serum_300,2
|
| 68 |
-
repaglinide_Steady,1
|
| 69 |
-
nateglinide_Steady,1
|
| 70 |
A1Cresult_Norm,1
|
| 71 |
-
|
| 72 |
-
|
| 73 |
-
|
| 74 |
-
|
| 75 |
-
|
| 76 |
-
|
| 77 |
-
|
| 78 |
-
|
| 79 |
-
|
| 80 |
-
|
| 81 |
-
|
|
|
|
|
|
|
| 82 |
metformin_Up,0
|
| 83 |
-
repaglinide_Up,0
|
| 84 |
nateglinide_No,0
|
| 85 |
nateglinide_Up,0
|
| 86 |
-
|
| 87 |
-
weight_75_100),0
|
| 88 |
-
weight_175_200),0
|
| 89 |
-
number_outpatient,0
|
| 90 |
-
race_Asian,0
|
| 91 |
-
chlorpropamide_Up,0
|
| 92 |
-
chlorpropamide_Steady,0
|
| 93 |
-
chlorpropamide_No,0
|
| 94 |
-
weight_0_25),0
|
| 95 |
-
age_very_young,0
|
| 96 |
-
gender_Unknown_Invalid,0
|
| 97 |
-
race_Hispanic,0
|
| 98 |
-
acetohexamide_Steady,0
|
| 99 |
-
diag_2_musculoskeletal,0
|
| 100 |
glyburide_Up,0
|
| 101 |
tolbutamide_Steady,0
|
| 102 |
-
|
| 103 |
-
|
| 104 |
-
|
| 105 |
-
|
| 106 |
rosiglitazone_Up,0
|
| 107 |
-
rosiglitazone_Steady,0
|
| 108 |
acarbose_No,0
|
| 109 |
-
acarbose_Steady,0
|
| 110 |
-
miglitol_Steady,0
|
| 111 |
-
miglitol_Up,0
|
| 112 |
acarbose_Up,0
|
| 113 |
miglitol_No,0
|
| 114 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 115 |
troglitazone_Steady,0
|
| 116 |
-
number_emergency,0
|
| 117 |
-
tolazamide_Up,0
|
| 118 |
-
glyburide_metformin_No,0
|
| 119 |
-
glyburide_metformin_Steady,0
|
| 120 |
-
glyburide_metformin_Up,0
|
| 121 |
glipizide_metformin_Steady,0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
feature,importance
|
| 2 |
+
discharge_disposition_id,427
|
| 3 |
+
diag_1_target_encoded,339
|
| 4 |
+
number_inpatient_mean,300
|
| 5 |
+
num_lab_procedures,284
|
| 6 |
+
diag_2_target_encoded,248
|
| 7 |
+
diag_3_target_encoded,248
|
| 8 |
+
number_inpatient_sum,244
|
| 9 |
+
med_diagnosis_interaction,171
|
| 10 |
+
num_medications,146
|
| 11 |
+
payer_code_is_missing,133
|
| 12 |
+
diag_1_cat_target_encoded,127
|
| 13 |
+
medical_specialty_target_encoded,116
|
| 14 |
+
time_in_hospital,112
|
| 15 |
+
number_inpatient_max,107
|
| 16 |
+
diag_2_cat_target_encoded,99
|
| 17 |
+
diag_3_cat_target_encoded,90
|
| 18 |
+
number_diagnoses,89
|
| 19 |
+
admission_type_id,78
|
| 20 |
+
weight_is_missing,76
|
| 21 |
+
admission_source_id,69
|
| 22 |
+
total_visits,68
|
| 23 |
+
payer_code_target_encoded,67
|
| 24 |
+
los_medication_interaction,61
|
| 25 |
+
age_target_encoded,59
|
| 26 |
+
care_utilization_risk_score,56
|
| 27 |
+
number_inpatient,54
|
| 28 |
+
medical_specialty_is_missing,51
|
| 29 |
+
age_bucket,50
|
| 30 |
+
num_procedures,49
|
| 31 |
+
race_Caucasian,31
|
| 32 |
+
diabetesMed_Yes,30
|
| 33 |
+
unique_diagnosis_categories,29
|
| 34 |
+
insulin_Steady,22
|
| 35 |
+
insulin_No,19
|
| 36 |
+
glipizide_No,18
|
| 37 |
+
A1Cresult_is_missing,12
|
| 38 |
+
metformin_Steady,12
|
| 39 |
+
race_Other,11
|
| 40 |
+
insulin_Up,11
|
| 41 |
+
weight_75_100),10
|
| 42 |
+
gender_Male,10
|
| 43 |
+
acarbose_Steady,9
|
| 44 |
+
change_No,8
|
| 45 |
+
medication_complexity_score,8
|
| 46 |
+
race_is_missing,8
|
| 47 |
+
weight_50_75),8
|
| 48 |
+
pioglitazone_No,8
|
| 49 |
+
glipizide_Steady,8
|
| 50 |
+
A1Cresult_8,7
|
| 51 |
+
race_Hispanic,7
|
| 52 |
+
rosiglitazone_No,7
|
| 53 |
+
metformin_No,7
|
| 54 |
+
repaglinide_Up,6
|
| 55 |
+
weight_100_125),5
|
| 56 |
+
bmi_underweight,4
|
| 57 |
+
max_glu_serum_Norm,4
|
| 58 |
+
glyburide_No,4
|
| 59 |
+
glimepiride_No,3
|
| 60 |
+
bmi_normal,3
|
| 61 |
+
pioglitazone_Steady,3
|
| 62 |
glipizide_Up,3
|
| 63 |
+
race_Asian,3
|
| 64 |
+
glimepiride_Steady,3
|
| 65 |
+
max_glu_serum_300,3
|
| 66 |
+
repaglinide_No,3
|
| 67 |
+
glimepiride_Up,2
|
| 68 |
+
rosiglitazone_Steady,2
|
| 69 |
+
glyburide_Steady,2
|
| 70 |
+
bmi_overweight,2
|
| 71 |
+
repaglinide_Steady,2
|
| 72 |
+
weight_125_150),1
|
|
|
|
|
|
|
|
|
|
| 73 |
A1Cresult_Norm,1
|
| 74 |
+
pioglitazone_Up,1
|
| 75 |
+
glyburide_metformin_No,1
|
| 76 |
+
nateglinide_Steady,1
|
| 77 |
+
number_emergency,0
|
| 78 |
+
bmi_obese,0
|
| 79 |
+
tolazamide_Steady,0
|
| 80 |
+
tolazamide_Up,0
|
| 81 |
+
glyburide_metformin_Steady,0
|
| 82 |
+
glyburide_metformin_Up,0
|
| 83 |
+
chlorpropamide_No,0
|
| 84 |
+
chlorpropamide_Steady,0
|
| 85 |
+
chlorpropamide_Up,0
|
| 86 |
+
acetohexamide_Steady,0
|
| 87 |
metformin_Up,0
|
|
|
|
| 88 |
nateglinide_No,0
|
| 89 |
nateglinide_Up,0
|
| 90 |
+
weight_25_50),0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 91 |
glyburide_Up,0
|
| 92 |
tolbutamide_Steady,0
|
| 93 |
+
gender_Unknown_Invalid,0
|
| 94 |
+
weight_0_25),0
|
| 95 |
+
weight_150_175),0
|
| 96 |
+
weight_175_200),0
|
| 97 |
rosiglitazone_Up,0
|
|
|
|
| 98 |
acarbose_No,0
|
|
|
|
|
|
|
|
|
|
| 99 |
acarbose_Up,0
|
| 100 |
miglitol_No,0
|
| 101 |
+
miglitol_Steady,0
|
| 102 |
+
miglitol_Up,0
|
| 103 |
+
number_outpatient_mean,0
|
| 104 |
+
number_outpatient_max,0
|
| 105 |
+
number_outpatient_sum,0
|
| 106 |
+
number_emergency_mean,0
|
| 107 |
+
number_emergency_max,0
|
| 108 |
+
number_emergency_sum,0
|
| 109 |
troglitazone_Steady,0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 110 |
glipizide_metformin_Steady,0
|
| 111 |
+
glimepiride_pioglitazone_Steady,0
|
| 112 |
+
metformin_rosiglitazone_Steady,0
|
| 113 |
+
metformin_pioglitazone_Steady,0
|
| 114 |
+
number_outpatient,0
|
feature_importance.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
gradient_boosting_metrics.json
CHANGED
|
@@ -1,20 +1,20 @@
|
|
| 1 |
{
|
| 2 |
-
"roc_auc": 0.
|
| 3 |
-
"pr_auc": 0.
|
| 4 |
-
"precision": 0.
|
| 5 |
-
"recall": 0.
|
| 6 |
-
"f1": 0.
|
| 7 |
-
"accuracy": 0.
|
| 8 |
-
"balanced_accuracy": 0.
|
| 9 |
-
"sensitivity": 0.
|
| 10 |
-
"specificity":
|
| 11 |
-
"ppv": 0.
|
| 12 |
-
"npv": 0.
|
| 13 |
-
"fpr": 0.
|
| 14 |
-
"fnr":
|
| 15 |
-
"true_positives":
|
| 16 |
-
"true_negatives":
|
| 17 |
-
"false_positives":
|
| 18 |
-
"false_negatives":
|
| 19 |
-
"brier_score": 0.
|
| 20 |
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"roc_auc": 0.8424324119878427,
|
| 3 |
+
"pr_auc": 0.3999922224076041,
|
| 4 |
+
"precision": 0.697841726618705,
|
| 5 |
+
"recall": 0.056924882629107984,
|
| 6 |
+
"f1": 0.10526315789473685,
|
| 7 |
+
"accuracy": 0.8919751064526695,
|
| 8 |
+
"balanced_accuracy": 0.5269138829486517,
|
| 9 |
+
"sensitivity": 0.056924882629107984,
|
| 10 |
+
"specificity": 0.9969028832681955,
|
| 11 |
+
"ppv": 0.697841726618705,
|
| 12 |
+
"npv": 0.8937590903080788,
|
| 13 |
+
"fpr": 0.003097116731804439,
|
| 14 |
+
"fnr": 0.943075117370892,
|
| 15 |
+
"true_positives": 97,
|
| 16 |
+
"true_negatives": 13519,
|
| 17 |
+
"false_positives": 42,
|
| 18 |
+
"false_negatives": 1607,
|
| 19 |
+
"brier_score": 0.0802369530229672
|
| 20 |
}
|
gradient_boosting_model.joblib
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:655b9aa5d82389fcd7fd94d9e72e0fa88278cf0c2c55afa86378e4a9494028b3
|
| 3 |
+
size 530372
|
learning_curves.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
metrics_comparison_across_folds.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
precision_recall_curve.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
roc_curve.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
training_summary.json
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
{
|
| 2 |
"model": "LightGBM Classifier",
|
| 3 |
"task": "Hospital 30-Day Readmission Risk Prediction",
|
| 4 |
-
"timestamp": "2025-12-
|
| 5 |
"environment": "kaggle",
|
| 6 |
"device": "gpu",
|
| 7 |
"evaluation_pipeline": {
|
|
@@ -15,50 +15,50 @@
|
|
| 15 |
"total_samples": 101766,
|
| 16 |
"development_size": 86501,
|
| 17 |
"final_test_size": 15265,
|
| 18 |
-
"n_features":
|
| 19 |
},
|
| 20 |
"best_params": {
|
| 21 |
"n_estimators": 150,
|
| 22 |
-
"learning_rate": 0.
|
| 23 |
-
"num_leaves":
|
| 24 |
"max_depth": -1,
|
| 25 |
"subsample": 0.7,
|
| 26 |
"colsample_bytree": 0.7,
|
| 27 |
-
"reg_alpha": 0.
|
| 28 |
"reg_lambda": 0.1
|
| 29 |
},
|
| 30 |
"cross_validation": {
|
| 31 |
-
"mean_roc_auc": 0.
|
| 32 |
-
"std_roc_auc": 0.
|
| 33 |
"fold_scores": [
|
| 34 |
-
0.
|
| 35 |
-
0.
|
| 36 |
-
0.
|
| 37 |
-
0.
|
| 38 |
-
0.
|
| 39 |
],
|
| 40 |
"n_folds": 5
|
| 41 |
},
|
| 42 |
"final_test_metrics": {
|
| 43 |
-
"roc_auc": 0.
|
| 44 |
-
"pr_auc": 0.
|
| 45 |
-
"precision": 0.
|
| 46 |
-
"recall": 0.
|
| 47 |
-
"f1": 0.
|
| 48 |
-
"accuracy": 0.
|
| 49 |
-
"balanced_accuracy": 0.
|
| 50 |
-
"sensitivity": 0.
|
| 51 |
-
"specificity":
|
| 52 |
-
"ppv": 0.
|
| 53 |
-
"npv": 0.
|
| 54 |
-
"fpr": 0.
|
| 55 |
-
"fnr":
|
| 56 |
-
"true_positives":
|
| 57 |
-
"true_negatives":
|
| 58 |
-
"false_positives":
|
| 59 |
-
"false_negatives":
|
| 60 |
-
"brier_score": 0.
|
| 61 |
},
|
| 62 |
-
"hyperparameter_search_time_seconds":
|
| 63 |
-
"total_time_seconds":
|
| 64 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"model": "LightGBM Classifier",
|
| 3 |
"task": "Hospital 30-Day Readmission Risk Prediction",
|
| 4 |
+
"timestamp": "2025-12-15 10:55:05",
|
| 5 |
"environment": "kaggle",
|
| 6 |
"device": "gpu",
|
| 7 |
"evaluation_pipeline": {
|
|
|
|
| 15 |
"total_samples": 101766,
|
| 16 |
"development_size": 86501,
|
| 17 |
"final_test_size": 15265,
|
| 18 |
+
"n_features": 113
|
| 19 |
},
|
| 20 |
"best_params": {
|
| 21 |
"n_estimators": 150,
|
| 22 |
+
"learning_rate": 0.05,
|
| 23 |
+
"num_leaves": 31,
|
| 24 |
"max_depth": -1,
|
| 25 |
"subsample": 0.7,
|
| 26 |
"colsample_bytree": 0.7,
|
| 27 |
+
"reg_alpha": 0.0,
|
| 28 |
"reg_lambda": 0.1
|
| 29 |
},
|
| 30 |
"cross_validation": {
|
| 31 |
+
"mean_roc_auc": 0.839936607760943,
|
| 32 |
+
"std_roc_auc": 0.005520489368125089,
|
| 33 |
"fold_scores": [
|
| 34 |
+
0.8422628166877643,
|
| 35 |
+
0.8423694634254875,
|
| 36 |
+
0.8372258386399722,
|
| 37 |
+
0.8318016530952919,
|
| 38 |
+
0.8460232669561988
|
| 39 |
],
|
| 40 |
"n_folds": 5
|
| 41 |
},
|
| 42 |
"final_test_metrics": {
|
| 43 |
+
"roc_auc": 0.8424324119878427,
|
| 44 |
+
"pr_auc": 0.3999922224076041,
|
| 45 |
+
"precision": 0.697841726618705,
|
| 46 |
+
"recall": 0.056924882629107984,
|
| 47 |
+
"f1": 0.10526315789473685,
|
| 48 |
+
"accuracy": 0.8919751064526695,
|
| 49 |
+
"balanced_accuracy": 0.5269138829486517,
|
| 50 |
+
"sensitivity": 0.056924882629107984,
|
| 51 |
+
"specificity": 0.9969028832681955,
|
| 52 |
+
"ppv": 0.697841726618705,
|
| 53 |
+
"npv": 0.8937590903080788,
|
| 54 |
+
"fpr": 0.003097116731804439,
|
| 55 |
+
"fnr": 0.943075117370892,
|
| 56 |
+
"true_positives": 97,
|
| 57 |
+
"true_negatives": 13519,
|
| 58 |
+
"false_positives": 42,
|
| 59 |
+
"false_negatives": 1607,
|
| 60 |
+
"brier_score": 0.0802369530229672
|
| 61 |
},
|
| 62 |
+
"hyperparameter_search_time_seconds": 7688.534896850586,
|
| 63 |
+
"total_time_seconds": 12862.338946819305
|
| 64 |
}
|
validation_curves.png
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|