Comparison of classical machine learning algorithms in the task of handwritten digits classification

Main Article Content

DOI

Oleksandr Voloshchenko

oleksandr.voloshchenko@pollub.edu.pl

Małgorzata Plechawska-Wójcik

m.plechawska@pollub.pl

Abstract

The purpose of this paper is to compare classical machine learning algorithms for handwritten number classification. The following algorithms were chosen for comparison: Logistic Regression, SVM, Decision Tree, Random Forest and k-NN. MNIST handwritten digit database is used in the task of training and testing the above algorithms. The dataset consists of 70,000 images of numbers from 0 to 9. The algorithms are compared considering such criteria as the learning speed, prediction construction speed, host machine load, and classification accuracy. Each algorithm went through the training and testing phases 100 times, with the desired KPIs retained at each iteration. The results were averaged to reach reliable outcomes.

Keywords:

machine learning, classification, MNIST, classical algorithms

References

Article Details

Voloshchenko, O., & Plechawska-Wójcik, M. (2021). Comparison of classical machine learning algorithms in the task of handwritten digits classification. Journal of Computer Sciences Institute, 21, 279–286. https://doi.org/10.35784/jcsi.2723