Classification of Headache Disorder Using Random Forest Algorithm

摘要

Headache disorder is one of the most often illness. At least 50% of the world’s population has experienced a headache. Primary headaches have several types; migraine, tension, cluster, and medication overuse. Computer aid for diagnosis could help people locate the headache type without the need to meet the doctor. The Random Forest algorithm was used in this study to produce a reliable model for classifying the headaches types and generate feature importance. In this study, the Migbase dataset was used, and several parameters of the algorithm were tuned to produce the best model. Based on the experiment results, the best accuracy reaching 99,56% with the Random Forest parameters are 100 for n_estimators, 33 for max_features, and 5 for max_depth.