prabinpanta0's picture
Update README.md
9c89bfe verified
---
license: mit
datasets:
- prabinpanta0/genki_hospital
pipeline_tag: text-classification
tags:
- patient-readmission
- predictive-model
- healthcare
- sklearn
- keras
- LightGBM
language:
- en
model_type: classification
task: patient-readmission-prediction
app_mode: standalone
frameworks:
- scikit-learn
- tensorflow
- LightGBM
metrics:
- accuracy
- precision
- recall
- f1
- roc_auc
---
# Patient Readmission Prediction
## Tranning
Github: [prabinpanta0/Patient-Readmission-Prediction](https://github.com/prabinpanta0/Patient-Readmission-Prediction)
## Dataset
* Original Source: [Kaggle/datasets/dubradave/hospital-readmissions](https://kaggle.com/datasets/dubradave/hospital-readmissions)
* Import Source: [HuggingFace/datasets/prabinpanta0/genki_hospital](https://huggingface.co/datasets/prabinpanta0/genki_hospital)
```Json
{
"model_id": "prabinpanta0/Patient-Readmission-Prediction",
"model_type": "sequence-classification",
"library": {
"random_forest": "scikit-learn",
"logistic_regression": "scikit-learn",
"k_nearest": "scikit-learn",
"svc": "scikit-learn",
"naive_bayes": "scikit-learn",
"neural_network": "keras",
"cross_validation_random_forest": "scikit-learn",
"cross_validation_logistic_regression": "scikit-learn",
"cross_validation_lightgbm": "LightGBM"
},
"model_architectures": {
"random_forest": "RandomForestClassifier",
"logistic_regression": "LogisticRegression",
"k_nearest": "KNeighborsClassifier",
"svc": "SVC",
"naive_bayes": "MultinomialNB",
"neural_network": "NeuralNetwork",
"cross_validation_random_forest": "RandomForestClassifier",
"cross_validation_logistic_regression": "LogisticRegression",
"cross_validation_lightgbm": "LGBMClassifier"
},
"model_paths": {
"random_forest": "model_RandomForestClassifier.pkl",
"logistic_regression": "model_Logistic_Regression.pkl",
"k_nearest": "model_K_nearest.pkl",
"svc": "model_svc.pkl",
"naive_bayes": "model_naive_bayes.pkl",
"neural_network": "neural_network.keras",
"cross_validation_random_forest": "model_rf.pkl",
"cross_validation_logistic_regression": "model_lr.pkl",
"cross_validation_lightgbm": "model_lgbm.pkl"
},
"model_classes": {
"random_forest": "RandomForestClassifier",
"logistic_regression": "LogisticRegression",
"k_nearest": "KNeighborsClassifier",
"svc": "SVC",
"naive_bayes": "MultinomialNB",
"neural_network": "NeuralNetwork",
"cross_validation_random_forest": "RandomForestClassifier",
"cross_validation_logistic_regression": "LogisticRegression"
},
"model_configs": {
"random_forest": {
"n_estimators": 100,
"max_depth": 5
},
"logistic_regression": {
"C": 1,
"max_iter": 1000
},
"k_nearest": {
"n_neighbors": 5
},
"svc": {
"C": 1,
"kernel": "linear"
},
"naive_bayes": {
"alpha": 1
},
"neural_network": {
"input_dim": 10,
"output_dim": 1,
"hidden_dim": 10
},
"cross_validation_random_forest": {
"n_estimators": 100,
"max_depth": 5
},
"cross_validation_logistic_regression": {
"C": 1,
"max_iter": 1000
},
"cross_validation_lightgbm": {
"random_state": 42
}
}
}
```
## metrics
|Model|Accuracy |Precision |Recall |AUC-ROC |
|-----|------------------|------------------|------------------|------------------|
|Random Forest|0.86544 |0.8734358240972471|0.8337883959044369|0.8635809449401703|
|Logistic Regression|0.74736 |0.7493540051679587|0.6928327645051194|0.7441573461079813|
|K-Nearest Neighbors|0.84112 |0.8543724844493231|0.7969283276450512|0.838524404786381 |
|Support Vector Classifier|0.84256 |0.8492462311557789|0.8075085324232082|0.8405012541634113|
|Naive Bayes|0.74176 |0.7692307692307693|0.6416382252559727|0.7358793535918418|
|Neural Network|0.87664 |0.889009009009009 |0.8419795221843004|0.8746042189234755|
|Random Forest (Cross-Validation)|0.86544 |0.8734358240972471|0.8337883959044369|0.8635809449401703|
|Logistic Regression (Cross-Validation)|0.74736 |0.7493540051679587|0.6928327645051194|0.7441573461079813|
|LightGBM (Cross-Validation)|0.8728 |0.8773418168964299|0.847098976109215 |0.8712904519100293|
|Random Forest|Logistic Regression|K-Nearest Neighbors|Support Vector Classifier|Naive Bayes |Neural Network |Random Forest (Cross-Validation)|Logistic Regression (Cross-Validation)|LightGBM (Cross-Validation)|
|-------------|-------------------|-------------------|-------------------------|------------------|------------------|--------------------------------|--------------------------------------|---------------------------|
|1.0 |0.7453866666666666 |0.8901866666666667 |0.8530133333333333 |0.7455466666666667|0.88288 |1.0 |0.7453866666666666 |0.9045866666666667 |
|1.0 |0.7449201741654572 |0.9005328596802842 |0.8556024378809189 |0.7743332882090158|0.8964114832535885|1.0 |0.7449201741654572 |0.910874897792314 |
|1.0 |0.6979827742520399 |0.8618540344514959 |0.8272892112420671 |0.6482320942883046|0.849274705349048 |1.0 |0.6979827742520399 |0.8837262012692656 |
|1.0 |0.7427552396345833 |0.8886139001594574 |0.8515853672571407 |0.7401446588709305|0.8810145438895148|1.0 |0.7427552396345833 |0.9034286859660855 |