A Comparative Study of Supervised Machine Learning Algorithms for Fruit Prediction

Abstract

In this paper, machine learning techniques have been applied for the fruit image classification and prediction over a large dataset. In the implemented work, five models have been developed and their performances are compared in predicting the fruit names. These models are based on five supervised learning techniques i.e., Support Vector Machine (SVM), Random Forest, K-Nearest Neighbor (KNN), Naive Bayes, and Softmax. The experimental results show that Support Vector Machine algorithm performs the best for large datasets and also Support Vector Machine is the best for small datasets. The results also reveal that reduction in the number of fruits reduces the accuracy’s of each algorithm.