Predicting Workplace Injuries Using Machine Learning Algorithms

Divya Sukumar, Ji Zhang, Xiaohui Tao, Xin Wang, Wenbin Zhang

Abstract: Predicting workplace injury using automated techniques opens newer possibilities in evidence-based research. This paper presents our preliminary research in a PhD project in predicting workplace incidents using machine learning algorithms. The analysis on the model performance using several mainstream machine learning algorithms including random forest, k-nearest neighbor and decision tree indicated that the general performance of the decision tree model was found to be statistically higher than that of the other two algorithms.