The Use of XGBoost Algorithm to Analyse the Severity of Traffic Accident Victims

Abstract

Traffic accidents are still significant contributors to a fairly high death. Denpasar’s resort police record every traffic accident in the form of a daily report. The stored data can generate valuable information to improve policies and propagate better traffic practices. This research utilizes the classification technique with the XGBoost, random forest algorithm, and SMOTE method. The study shows that the SMOTE technique can increase the model's accuracy. Using the classification method with the two algorithms produces factors that affect the severity of traffic accident victims with feature importance. The feature importance obtained using the XGBoost model by counting the weight value for testing using the original dataset, the dataset for the type of two-wheeled vehicle, and the dataset of the kind of vehicle other than two-wheeled indicate that the variables influencing the severity of victims in road accidents are the time of accident between 00.00-06.00, the type of vehicle motorcycle, the type of opponent vehicle truck and pickup car, the age of the driver between 16-25, sub-district road status and front – side type of accident.