Effect of clustering data in improving machine learning model accuracy

Samih M. Mostafa, Hirofumi Amano

Research output: Contribution to journalArticlepeer-review

10 Citations (Scopus)


Supervised machine learning algorithms consider the relationship between dependent and independent variables rather than the relationship between the instances. Machine learning algorithms try to learn the relationship between the input and output from the historical data in order to attain precise predictions about unseen future. Conventional foretelling algorithms are usually based on a model learned and trained from historical data. The instances in the historical data may vary in its characteristics. The variation may be a result of difference in case's pertinence degree to some cases compared to others. However, the problem with such machine learning algorithms is their dealing with the whole data without considering this variation. This paper presents a novel technique to the trained model to improve the prediction accuracy. The proposed method clusters the data using K-means clustering algorithm, and then applies the prediction algorithm to every cluster. The value of K which gives the highest accuracy is selected. The authors performed comparative study of the proposed technique and popular prediction methods namely Linear Regression, Ridge, Lasso, and Elastic. On analysing on five datasets with different sizes and different number of clusters, it was observed that the accuracy of the proposed technique is better from the point of view of Root Mean Square Error (RMSE), and coefficient of determination (Rz).

Original languageEnglish
Pages (from-to)2973-2981
Number of pages9
JournalJournal of Theoretical and Applied Information Technology
Issue number21
Publication statusPublished - Nov 15 2019
Externally publishedYes

All Science Journal Classification (ASJC) codes

  • Theoretical Computer Science
  • General Computer Science


Dive into the research topics of 'Effect of clustering data in improving machine learning model accuracy'. Together they form a unique fingerprint.

Cite this