\n", + " | age | \n", + "anaemia | \n", + "creatinine_phosphokinase | \n", + "diabetes | \n", + "ejection_fraction | \n", + "high_blood_pressure | \n", + "platelets | \n", + "serum_creatinine | \n", + "serum_sodium | \n", + "sex | \n", + "smoking | \n", + "time | \n", + "DEATH_EVENT | \n", + "
---|---|---|---|---|---|---|---|---|---|---|---|---|---|
0 | \n", + "75.0 | \n", + "0 | \n", + "582 | \n", + "0 | \n", + "20 | \n", + "1 | \n", + "265000.00 | \n", + "1.9 | \n", + "130 | \n", + "1 | \n", + "0 | \n", + "4 | \n", + "1 | \n", + "
1 | \n", + "55.0 | \n", + "0 | \n", + "7861 | \n", + "0 | \n", + "38 | \n", + "0 | \n", + "263358.03 | \n", + "1.1 | \n", + "136 | \n", + "1 | \n", + "0 | \n", + "6 | \n", + "1 | \n", + "
2 | \n", + "65.0 | \n", + "0 | \n", + "146 | \n", + "0 | \n", + "20 | \n", + "0 | \n", + "162000.00 | \n", + "1.3 | \n", + "129 | \n", + "1 | \n", + "1 | \n", + "7 | \n", + "1 | \n", + "
3 | \n", + "50.0 | \n", + "1 | \n", + "111 | \n", + "0 | \n", + "20 | \n", + "0 | \n", + "210000.00 | \n", + "1.9 | \n", + "137 | \n", + "1 | \n", + "0 | \n", + "7 | \n", + "1 | \n", + "
4 | \n", + "65.0 | \n", + "1 | \n", + "160 | \n", + "1 | \n", + "20 | \n", + "0 | \n", + "327000.00 | \n", + "2.7 | \n", + "116 | \n", + "0 | \n", + "0 | \n", + "8 | \n", + "1 | \n", + "
... | \n", + "... | \n", + "... | \n", + "... | \n", + "... | \n", + "... | \n", + "... | \n", + "... | \n", + "... | \n", + "... | \n", + "... | \n", + "... | \n", + "... | \n", + "... | \n", + "
294 | \n", + "62.0 | \n", + "0 | \n", + "61 | \n", + "1 | \n", + "38 | \n", + "1 | \n", + "155000.00 | \n", + "1.1 | \n", + "143 | \n", + "1 | \n", + "1 | \n", + "270 | \n", + "0 | \n", + "
295 | \n", + "55.0 | \n", + "0 | \n", + "1820 | \n", + "0 | \n", + "38 | \n", + "0 | \n", + "270000.00 | \n", + "1.2 | \n", + "139 | \n", + "0 | \n", + "0 | \n", + "271 | \n", + "0 | \n", + "
296 | \n", + "45.0 | \n", + "0 | \n", + "2060 | \n", + "1 | \n", + "60 | \n", + "0 | \n", + "742000.00 | \n", + "0.8 | \n", + "138 | \n", + "0 | \n", + "0 | \n", + "278 | \n", + "0 | \n", + "
297 | \n", + "45.0 | \n", + "0 | \n", + "2413 | \n", + "0 | \n", + "38 | \n", + "0 | \n", + "140000.00 | \n", + "1.4 | \n", + "140 | \n", + "1 | \n", + "1 | \n", + "280 | \n", + "0 | \n", + "
298 | \n", + "50.0 | \n", + "0 | \n", + "196 | \n", + "0 | \n", + "45 | \n", + "0 | \n", + "395000.00 | \n", + "1.6 | \n", + "136 | \n", + "1 | \n", + "1 | \n", + "285 | \n", + "0 | \n", + "
299 rows × 13 columns
\n", + "\n", + " | age | \n", + "anaemia | \n", + "creatinine_phosphokinase | \n", + "diabetes | \n", + "ejection_fraction | \n", + "high_blood_pressure | \n", + "platelets | \n", + "serum_creatinine | \n", + "serum_sodium | \n", + "sex | \n", + "smoking | \n", + "time | \n", + "DEATH_EVENT | \n", + "
---|---|---|---|---|---|---|---|---|---|---|---|---|---|
0 | \n", + "75.0 | \n", + "Non-Anaemic | \n", + "582 | \n", + "Non-Diabetic | \n", + "20 | \n", + "High | \n", + "265000.00 | \n", + "1.9 | \n", + "130 | \n", + "Male | \n", + "Non-Smoker | \n", + "4 | \n", + "Deceased | \n", + "
1 | \n", + "55.0 | \n", + "Non-Anaemic | \n", + "7861 | \n", + "Non-Diabetic | \n", + "38 | \n", + "Normal | \n", + "263358.03 | \n", + "1.1 | \n", + "136 | \n", + "Male | \n", + "Non-Smoker | \n", + "6 | \n", + "Deceased | \n", + "
2 | \n", + "65.0 | \n", + "Non-Anaemic | \n", + "146 | \n", + "Non-Diabetic | \n", + "20 | \n", + "Normal | \n", + "162000.00 | \n", + "1.3 | \n", + "129 | \n", + "Male | \n", + "Smoker | \n", + "7 | \n", + "Deceased | \n", + "
3 | \n", + "50.0 | \n", + "Anaemic | \n", + "111 | \n", + "Non-Diabetic | \n", + "20 | \n", + "Normal | \n", + "210000.00 | \n", + "1.9 | \n", + "137 | \n", + "Male | \n", + "Non-Smoker | \n", + "7 | \n", + "Deceased | \n", + "
4 | \n", + "65.0 | \n", + "Anaemic | \n", + "160 | \n", + "Diabetic | \n", + "20 | \n", + "Normal | \n", + "327000.00 | \n", + "2.7 | \n", + "116 | \n", + "Female | \n", + "Non-Smoker | \n", + "8 | \n", + "Deceased | \n", + "
\n", + " | count | \n", + "mean | \n", + "std | \n", + "min | \n", + "25% | \n", + "50% | \n", + "75% | \n", + "max | \n", + "
---|---|---|---|---|---|---|---|---|
age | \n", + "299.000000 | \n", + "60.833893 | \n", + "11.894809 | \n", + "40.000000 | \n", + "51.000000 | \n", + "60.000000 | \n", + "70.000000 | \n", + "95.000000 | \n", + "
anaemia | \n", + "299.000000 | \n", + "0.431438 | \n", + "0.496107 | \n", + "0.000000 | \n", + "0.000000 | \n", + "0.000000 | \n", + "1.000000 | \n", + "1.000000 | \n", + "
creatinine_phosphokinase | \n", + "299.000000 | \n", + "581.839465 | \n", + "970.287881 | \n", + "23.000000 | \n", + "116.500000 | \n", + "250.000000 | \n", + "582.000000 | \n", + "7861.000000 | \n", + "
diabetes | \n", + "299.000000 | \n", + "0.418060 | \n", + "0.494067 | \n", + "0.000000 | \n", + "0.000000 | \n", + "0.000000 | \n", + "1.000000 | \n", + "1.000000 | \n", + "
ejection_fraction | \n", + "299.000000 | \n", + "38.083612 | \n", + "11.834841 | \n", + "14.000000 | \n", + "30.000000 | \n", + "38.000000 | \n", + "45.000000 | \n", + "80.000000 | \n", + "
high_blood_pressure | \n", + "299.000000 | \n", + "0.351171 | \n", + "0.478136 | \n", + "0.000000 | \n", + "0.000000 | \n", + "0.000000 | \n", + "1.000000 | \n", + "1.000000 | \n", + "
platelets | \n", + "299.000000 | \n", + "263358.029264 | \n", + "97804.236869 | \n", + "25100.000000 | \n", + "212500.000000 | \n", + "262000.000000 | \n", + "303500.000000 | \n", + "850000.000000 | \n", + "
serum_creatinine | \n", + "299.000000 | \n", + "1.393880 | \n", + "1.034510 | \n", + "0.500000 | \n", + "0.900000 | \n", + "1.100000 | \n", + "1.400000 | \n", + "9.400000 | \n", + "
serum_sodium | \n", + "299.000000 | \n", + "136.625418 | \n", + "4.412477 | \n", + "113.000000 | \n", + "134.000000 | \n", + "137.000000 | \n", + "140.000000 | \n", + "148.000000 | \n", + "
sex | \n", + "299.000000 | \n", + "0.648829 | \n", + "0.478136 | \n", + "0.000000 | \n", + "0.000000 | \n", + "1.000000 | \n", + "1.000000 | \n", + "1.000000 | \n", + "
smoking | \n", + "299.000000 | \n", + "0.321070 | \n", + "0.467670 | \n", + "0.000000 | \n", + "0.000000 | \n", + "0.000000 | \n", + "1.000000 | \n", + "1.000000 | \n", + "
time | \n", + "299.000000 | \n", + "130.260870 | \n", + "77.614208 | \n", + "4.000000 | \n", + "73.000000 | \n", + "115.000000 | \n", + "203.000000 | \n", + "285.000000 | \n", + "
DEATH_EVENT | \n", + "299.000000 | \n", + "0.321070 | \n", + "0.467670 | \n", + "0.000000 | \n", + "0.000000 | \n", + "0.000000 | \n", + "1.000000 | \n", + "1.000000 | \n", + "
\n", + " | F1 Score | \n", + "Accuracy | \n", + "Precision | \n", + "Recall | \n", + "Classification Error | \n", + "
---|---|---|---|---|---|
Naive Bayes | \n", + "0.729730 | \n", + "0.756098 | \n", + "0.900000 | \n", + "0.613636 | \n", + "0.243902 | \n", + "
Logistic Regression | \n", + "0.800000 | \n", + "0.792683 | \n", + "0.829268 | \n", + "0.772727 | \n", + "0.207317 | \n", + "
KNN | \n", + "0.813953 | \n", + "0.804878 | \n", + "0.833333 | \n", + "0.795455 | \n", + "0.195122 | \n", + "
SVM | \n", + "0.857143 | \n", + "0.853659 | \n", + "0.900000 | \n", + "0.818182 | \n", + "0.146341 | \n", + "
Decision Tree | \n", + "0.873563 | \n", + "0.865854 | \n", + "0.883721 | \n", + "0.863636 | \n", + "0.134146 | \n", + "
Random Forest | \n", + "0.917647 | \n", + "0.914634 | \n", + "0.951220 | \n", + "0.886364 | \n", + "0.085366 | \n", + "
XGBoost | \n", + "0.919540 | \n", + "0.914634 | \n", + "0.930233 | \n", + "0.909091 | \n", + "0.085366 | \n", + "
LightGBM | \n", + "0.955556 | \n", + "0.951220 | \n", + "0.934783 | \n", + "0.977273 | \n", + "0.048780 | \n", + "
\n", + " | F1 Score | \n", + "Accuracy | \n", + "Precision | \n", + "Recall | \n", + "Classification Error | \n", + "
---|---|---|---|---|---|
Random Forest | \n", + "0.930233 | \n", + "0.926829 | \n", + "0.952381 | \n", + "0.909091 | \n", + "0.073171 | \n", + "
LightGBM | \n", + "0.930233 | \n", + "0.926829 | \n", + "0.952381 | \n", + "0.909091 | \n", + "0.073171 | \n", + "
XGBoost | \n", + "0.955556 | \n", + "0.951220 | \n", + "0.934783 | \n", + "0.977273 | \n", + "0.048780 | \n", + "
Parameters | \n", + "Original | \n", + "Hypertuned | \n", + "
---|---|---|
F1 Score | \n", + "91.95%\n", + " | 95.55%\n", + " |
Accuracy | \n", + "91.46%\n", + " | 95.12%\n", + " |
Precision | \n", + "93.02%\n", + " | 93.47%\n", + " |
Recall | \n", + "90.90%\n", + " | 97.72%\n", + " |
Classification error | \n", + "8.535\n", + " | 4.87%\n", + " |
Parameters | \n", + "Original | \n", + "Hypertuned | \n", + "
---|---|---|
F1 Score | \n", + "95.55%\n", + " | 94.11%\n", + " |
Accuracy | \n", + "95.12%\n", + " | 93.90%\n", + " |
Precision | \n", + "93.47%\n", + " | 97.56%\n", + " |
Recall | \n", + "97.72%\n", + " | 90.90%\n", + " |
Classification error | \n", + "4.87%\n", + " | 6.09%\n", + " |
Parameters | \n", + "Original | \n", + "Hypertuned | \n", + "
---|---|---|
F1 Score | \n", + "91.76%\n", + " | 93.02%\n", + " |
Accuracy | \n", + "91.46%\n", + " | 92.68%\n", + " |
Precision | \n", + "95.12%\n", + " | 95.23%\n", + " |
Recall | \n", + "88.63%\n", + " | 90.90%\n", + " |
Classification error | \n", + "8.53%\n", + " | 7.31%\n", + " |
\n", + " | age | \n", + "anaemia | \n", + "creatinine_phosphokinase | \n", + "diabetes | \n", + "ejection_fraction | \n", + "high_blood_pressure | \n", + "platelets | \n", + "serum_creatinine | \n", + "serum_sodium | \n", + "sex | \n", + "smoking | \n", + "time | \n", + "DEATH_EVENT | \n", + "
---|---|---|---|---|---|---|---|---|---|---|---|---|---|
279 | \n", + "55.0 | \n", + "0 | \n", + "84 | \n", + "1 | \n", + "38 | \n", + "0 | \n", + "451000.00 | \n", + "1.3 | \n", + "136 | \n", + "0 | \n", + "0 | \n", + "246 | \n", + "0 | \n", + "
280 | \n", + "70.0 | \n", + "0 | \n", + "2695 | \n", + "1 | \n", + "40 | \n", + "0 | \n", + "241000.00 | \n", + "1.0 | \n", + "137 | \n", + "1 | \n", + "0 | \n", + "247 | \n", + "0 | \n", + "
281 | \n", + "70.0 | \n", + "0 | \n", + "582 | \n", + "0 | \n", + "40 | \n", + "0 | \n", + "51000.00 | \n", + "2.7 | \n", + "136 | \n", + "1 | \n", + "1 | \n", + "250 | \n", + "0 | \n", + "
282 | \n", + "42.0 | \n", + "0 | \n", + "64 | \n", + "0 | \n", + "30 | \n", + "0 | \n", + "215000.00 | \n", + "3.8 | \n", + "128 | \n", + "1 | \n", + "1 | \n", + "250 | \n", + "0 | \n", + "
283 | \n", + "65.0 | \n", + "0 | \n", + "1688 | \n", + "0 | \n", + "38 | \n", + "0 | \n", + "263358.03 | \n", + "1.1 | \n", + "138 | \n", + "1 | \n", + "1 | \n", + "250 | \n", + "0 | \n", + "
284 | \n", + "50.0 | \n", + "1 | \n", + "54 | \n", + "0 | \n", + "40 | \n", + "0 | \n", + "279000.00 | \n", + "0.8 | \n", + "141 | \n", + "1 | \n", + "0 | \n", + "250 | \n", + "0 | \n", + "
285 | \n", + "55.0 | \n", + "1 | \n", + "170 | \n", + "1 | \n", + "40 | \n", + "0 | \n", + "336000.00 | \n", + "1.2 | \n", + "135 | \n", + "1 | \n", + "0 | \n", + "250 | \n", + "0 | \n", + "
286 | \n", + "60.0 | \n", + "0 | \n", + "253 | \n", + "0 | \n", + "35 | \n", + "0 | \n", + "279000.00 | \n", + "1.7 | \n", + "140 | \n", + "1 | \n", + "0 | \n", + "250 | \n", + "0 | \n", + "
287 | \n", + "45.0 | \n", + "0 | \n", + "582 | \n", + "1 | \n", + "55 | \n", + "0 | \n", + "543000.00 | \n", + "1.0 | \n", + "132 | \n", + "0 | \n", + "0 | \n", + "250 | \n", + "0 | \n", + "
288 | \n", + "65.0 | \n", + "0 | \n", + "892 | \n", + "1 | \n", + "35 | \n", + "0 | \n", + "263358.03 | \n", + "1.1 | \n", + "142 | \n", + "0 | \n", + "0 | \n", + "256 | \n", + "0 | \n", + "
289 | \n", + "90.0 | \n", + "1 | \n", + "337 | \n", + "0 | \n", + "38 | \n", + "0 | \n", + "390000.00 | \n", + "0.9 | \n", + "144 | \n", + "0 | \n", + "0 | \n", + "256 | \n", + "0 | \n", + "
290 | \n", + "45.0 | \n", + "0 | \n", + "615 | \n", + "1 | \n", + "55 | \n", + "0 | \n", + "222000.00 | \n", + "0.8 | \n", + "141 | \n", + "0 | \n", + "0 | \n", + "257 | \n", + "0 | \n", + "
291 | \n", + "60.0 | \n", + "0 | \n", + "320 | \n", + "0 | \n", + "35 | \n", + "0 | \n", + "133000.00 | \n", + "1.4 | \n", + "139 | \n", + "1 | \n", + "0 | \n", + "258 | \n", + "0 | \n", + "
292 | \n", + "52.0 | \n", + "0 | \n", + "190 | \n", + "1 | \n", + "38 | \n", + "0 | \n", + "382000.00 | \n", + "1.0 | \n", + "140 | \n", + "1 | \n", + "1 | \n", + "258 | \n", + "0 | \n", + "
293 | \n", + "63.0 | \n", + "1 | \n", + "103 | \n", + "1 | \n", + "35 | \n", + "0 | \n", + "179000.00 | \n", + "0.9 | \n", + "136 | \n", + "1 | \n", + "1 | \n", + "270 | \n", + "0 | \n", + "
294 | \n", + "62.0 | \n", + "0 | \n", + "61 | \n", + "1 | \n", + "38 | \n", + "1 | \n", + "155000.00 | \n", + "1.1 | \n", + "143 | \n", + "1 | \n", + "1 | \n", + "270 | \n", + "0 | \n", + "
295 | \n", + "55.0 | \n", + "0 | \n", + "1820 | \n", + "0 | \n", + "38 | \n", + "0 | \n", + "270000.00 | \n", + "1.2 | \n", + "139 | \n", + "0 | \n", + "0 | \n", + "271 | \n", + "0 | \n", + "
296 | \n", + "45.0 | \n", + "0 | \n", + "2060 | \n", + "1 | \n", + "60 | \n", + "0 | \n", + "742000.00 | \n", + "0.8 | \n", + "138 | \n", + "0 | \n", + "0 | \n", + "278 | \n", + "0 | \n", + "
297 | \n", + "45.0 | \n", + "0 | \n", + "2413 | \n", + "0 | \n", + "38 | \n", + "0 | \n", + "140000.00 | \n", + "1.4 | \n", + "140 | \n", + "1 | \n", + "1 | \n", + "280 | \n", + "0 | \n", + "
298 | \n", + "50.0 | \n", + "0 | \n", + "196 | \n", + "0 | \n", + "45 | \n", + "0 | \n", + "395000.00 | \n", + "1.6 | \n", + "136 | \n", + "1 | \n", + "1 | \n", + "285 | \n", + "0 | \n", + "