Journal of Computer Sciences Institute (Dec 2021)

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

  • Oleksandr Voloshchenko,
  • Małgorzata Plechawska-Wójcik

DOI
https://doi.org/10.35784/jcsi.2723
Journal volume & issue
Vol. 21

Abstract

Read online

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