Building of Informatics, Technology and Science
Vol 8 No 1 (2026): June 2026

Symptom-Based Classification of Migraine Severity Using Random Forest and LassoNet

Syehan Fariz Gustomo (Telkom University, Bandung)
Kemas Muslim Lhaksmana (Telkom University, Bandung)



Article Info

Publish Date
30 Jun 2026

Abstract

This study aims to develop and compare classification models for predicting symptom-based migraine intensity using a public dataset from Kaggle. The research was conducted on multiclass data with an imbalanced class distribution. Therefore, target creation and result interpretation must be carefully designed so that the resulting evaluation remains consistent with the characteristics of the data used. In this study, the Intensity variable was recoded into three operational classes: low, moderate, and high. The features used included symptoms, characteristics of migraine episodes, and `symptom_count`, which represents the number of symptoms in each sample. The two models compared were Random Forest and LassoNet, both of which were tested using Stratified 5-Fold Cross-Validation. Model performance was assessed using the Macro F1-score as the primary metric, Balanced Accuracy as the main supplementary metric, and Accuracy as a complementary metric. The test results showed that Random Forest performed better, with a Macro F1-score of 0.6366, Balanced Accuracy of 0.6181, and Accuracy of 0.6225. Meanwhile, LassoNet achieved a Macro F1-score of 0.2474, a Balanced Accuracy of 0.3333, and an Accuracy of 0.5900. These results indicate that symptom patterns in the dataset can still be utilized to distinguish migraine intensity within a computational classification framework, although the separation between closely related classes is not yet fully robust.

Copyrights © 2026






Journal Info

Abbrev

bits

Publisher

Subject

Computer Science & IT

Description

Building of Informatics, Technology and Science (BITS) is an open access media in publishing scientific articles that contain the results of research in information technology and computers. Paper that enters this journal will be checked for plagiarism and peer-rewiew first to maintain its quality. ...