Student Dropout Risk Classification: A Comparison of Random Forest and Naïve Bayes Algorithms
##plugins.themes.academic_pro.article.main##
Abstract
Student dropout challenges higher education, while class imbalance can obscure models’ ability to identify students at risk. This study compared Random Forest and Naïve Bayes using academic and nonacademic variables, emphasizing performance on the dropout class. Evaluation used 1,900 test records: 1,452 non-dropout and 448 dropout cases. Confusion matrices, classification metrics, a majority-class baseline, balanced accuracy, and Matthews correlation coefficient were examined. Naïve Bayes achieved 76.58% accuracy, exceeding Random Forest’s 73.21% but barely surpassing the 76.42% baseline. For dropout detection, Random Forest achieved 0.20 recall and a 0.26 F1-score, compared with 0.05 and 0.09 for Naïve Bayes. Random Forest identified 91 dropout cases versus 22, with approximately 1.93 additional false alarms per additional case detected. Despite its relative advantage in detecting dropout, Random Forest missed 357 of 448 cases. Neither model is suitable as a primary early warning classifier without further improved handling of class imbalance and careful decision thresholds.
##plugins.themes.academic_pro.article.details##

This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.
References
Bulut, O., Wongvorachan, T., He, S., & Lee, S. (2024). Enhancing high-school dropout identification: a collaborative approach integrating human and machine insights. Discover Education, 3(1). https://doi.org/10.1007/s44217-024-00209-4
Dikananda, A. R., Ali, I., Fathurrohman, Ade Rinaldi, R., & Iin. (2021). Genre e-sport gaming tournament classification using machine learning technique based on decision tree, Naïve Bayes, and random forest algorithm. IOP Conference Series: Materials Science and Engineering, 1088(1), 012037. https://doi.org/10.1088/1757-899x/1088/1/012037
Fansyuri, M., & Yunita, D. (2025). Analisis Kinerja Algoritma Naive Bayes dalam Memprediksi Kelulusan Mahasiswa Menggunakan Python. Jurnal Pustaka AI (Pusat Akses Kajian Teknologi Artificial Intelligence), 5(3), 716–723. https://doi.org/10.55382/jurnalpustakaai.v5i3.1491
Forest, R., Students, P., Success, A., Machine, U. C. I., Repository, L., Dropout, T., Forest, R., Validation, F. C., Curve, A. U., Coefficient, M. C., Forest, R., Auc, N., Forest, R., Kunci, K., & Forest, R. (2026). Klasifikasi Risiko Dropout Mahasiswa Menggunakan Algoritma Random Forest pada Dataset Predict Students ’ Dropout and Academic Success Novita Sari Siagian1 , Monica Sari Batubara 2 , Leony Sinaga3 1Program Studi Ilmu Komputer , Fakultas Matematika dan Ilmu. 10(2), 629–640.
Meiriza, A., Lestari, E., Putra, P., Monaputri, A., & Lestari, D. A. (2020). Prediction graduate student use naive bayes classifier. Sriwijaya International Conference on Information Technology and Its Applications (SICONIAN 2019), 370–375.
Niyogisubizo, J., Liao, L., Nziyumva, E., Murwanashyaka, E., & Nshimyumukiza, P. C. (2022). Predicting student’s dropout in university classes using two-layer ensemble machine learning approach: A novel stacked generalization. Computers and Education: Artificial Intelligence, 3, 100066.
Prasetya, R., & Ridwan, A. (2019). Data Mining Application on Weather Prediction Using Classification Tree, Naïve Bayes and K-Nearest Neighbor Algorithm With Model Testing of Supervised Learning Probabilistic Brier Score, Confusion Matrix and ROC. In JAICT Journal of Applied Communication and Information Technologies (Vol. 4, Number 2).
Rotifa, A. N., Alfino, F. T., Wardhana, P. C., & Putri, A. S. A. (2026). PERBANDINGAN MODEL MACHINE LEARNING UNTUK PREDIKSI DROP-OUT MAHASISWA DENGAN ANALISIS FEATURE IMPORTANCE BERBASIS STUDI LITERATUR. 10(4), 5595–5605.
Stinebrickner, R., & Stinebrickner, T. (2014). Academic performance and college dropout. Journal of Labor Economics, 32(3), 601–644.
Sumitro, R., & Ismail, J. (2026). JUTEK – JURNAL TEKNOLOGI Komparasi Algoritma K-Nearest Neighbor , Naïve Bayes , dan Random Forest dengan SMOTE untuk Prediksi Dropout Mahasiswa. 3(1), 18–25.
Wijaya, V. Y. D., & Brotosaputro, G. (2025). Penerapan Data Mining Dalam Prediksi Kinerja Akademik Mahasiswa Menggunakan Algoritma Machine Learning. Jurnal Infomedia: Teknik Informatika, Multimedia, Dan Jaringan, 10(2), 134–142.
Yuan, J., Qiu, X., Wu, J., Guo, J., Li, W., & Wang, Y.-G. (2024). Integrating behavior analysis with machine learning to predict online learning performance: A scientometric review and empirical study. http://arxiv.org/abs/2406.11847
Zalina Ramadhany, M., & Wati, L. (2026). Klasifikasi Mahasiswa Berpotensi Drop Out (Do) Menggunakan Algoritma Random Forest. JATI (Jurnal Mahasiswa Teknik Informatika), 10(2), 3543–3548. https://doi.org/10.36040/jati.v10i2.17666
Zulkifli, R., Andryadi, A. A., Nurfadhilah, D. S., Utami, L. L., Universitas, I., Al-ghifari, S. I. U., Langlangbuana, I. U., & Kunci, K. (2026). PENERAPAN ALGORITMA NAIVE BAYES DALAM MEMPREDIKSI MINAT MAHASISWA BARU Informatika Universitas Siliwangi Abstraksi Keywords : Pendahuluan Tinjauan Pustaka Data mining adalah proses penemuan pola dan. Journal of Information System Management (JOISM), 7(2), 252–259.