返回
Probabilistic Ensemble Framework for Injury Narrative Classification
DOI:10.3390/ai5030082.png)
摘要
En 中文
In this research, we analyzed narratives from the National Electronic Injury Surveillance System (NEISS) dataset to predict the top two injury codes using a comparative study of ensemble machine learning (ML) models. Four ensemble models were evaluated: Random Forest (RF) combined with Logistic Regression (LR), K-Nearest Neighbor (KNN) paired with RF, LR combined with KNN, and a model integrating LR, RF, and KNN, all utilizing a probabilistic likelihood-based approach to improve decision-making across different classifiers. The combined KNN + LR ensemble achieved an accuracy of 90.47% for the top one prediction, while the KNN + RF + LR model excelled in predicting the top two injury codes with a very high accuracy of 99.50%. These results demonstrate the significant potential of ensemble models to enhance unstructured narrative classification accuracy, particularly in addressing underrepresented cases, and the potential of the proposed probabilistic ensemble framework ML models in improving decision-making in public health and safety, providing a foundation for future research in automated clinical narrative classification and predictive modeling, especially in scenarios with imbalanced data.
Keyword:
injury classification
injury narratives
machine learning models
ensemble models
期刊
A
IF:
5
论文数:
1.1K
被引数:
941
机构
引用论文
Onset of draw resonance during isothermal melt spinning: A comparison between measurements and predictions等温熔体纺丝过程中拉伸共振的开始: 测量与预测之间的比较

