Related Experiment Video
Updated: Oct 26, 2025
A Machine Learning Approach to Design an Efficient Selective Screening of Mild Cognitive Impairment
Published on: January 11, 2020
Comparison of Supervised Machine Learning Algorithms for Classifying of Home Discharge Possibility in Convalescent
Takeshi Imura1, Haruki Toda2, Yuji Iwamoto3
1Department of Rehabilitation, Faculty of Health Sciences, Hiroshima Cosmopolitan University, 3-2-1, Otsuka-Higashi, Hiroshima 731-3166, Japan; Department of Rehabilitation, Araki Neurosurgical Hospital, Hiroshima, Japan; Graduate School of Humanities and Social Sciences, Hiroshima University, Hiroshima, Japan.
Objectives:
Classifying the possibility of home discharge is important during stroke rehabilitation to support decision-making. There have been several studies on supervised machine learning algorithms, but only a few have compared the performance of different algorithms based on the same dataset for the classification of home discharge possibility. Therefore, we aimed to evaluate five supervised machine learning algorithms for the classification of home discharge possibility in stroke patients.
Materials And Methods:
This was a secondary analysis based on the data of 481 stroke patients from the database of our institution. Five models developed by supervised machine learning algorithms, including decision tree (DT), linear discriminant analysis (LDA), k-nearest neighbors (k-NN), support vector machine (SVM), and random forest (RF) were compared by constructing a classification system based on the same dataset. Several parameters including classification accuracy, area under the curve (AUC), and F1 score (a weighted average of precision and recall) were used for model evaluation.
Results:
The k-NN model had the best classification accuracy (84.0%) with a moderate AUC (0.88) and F1 score (87.8). The SVM model also showed high classification accuracy (82.6%) along with the highest AUC (0.91), sensitivity (94.4), negative predictive value (87.5), and negative likelihood ratio (0.088). The DT, LDA, and RF models had high classification accuracies (≥ 79.9%) with moderate AUCs (≥ 0.84) and F1 scores (≥ 83.8).
Conclusions:
Regarding model performance, the k-NN and SVM seemed the best candidate algorithms for classifying the possibility of home discharge in stroke patients.
