Binary-Class Tabular Model for Pumpkin Seeds Identification Using Python and Scikit-Learn

Template Credit: Adapted from a template made available by Dr. Jason Brownlee of Machine Learning Mastery.

SUMMARY: The project aims to construct a predictive model using various machine learning algorithms and document the end-to-end steps using a template. The Pumpkin Seeds Identification dataset is a binary-class modeling situation where we attempt to predict one of two possible outcomes.

INTRODUCTION: Pumpkin seeds are frequently consumed as confection worldwide because of their adequate amount of protein, fat, carbohydrate, and mineral contents. The research team carried out a study on the two most important types of pumpkin seed, “Ürgüp Sivrisi” and “Çerçevelik,” generally grown in Ürgüp and Karacaören regions in Turkey. Furthermore, the morphological measurements of 2500 pumpkin seeds of both varieties were captured using the gray and binary forms of threshold techniques.

ANALYSIS: The average performance of the machine learning algorithms achieved an accuracy benchmark of 87.06% after training. Furthermore, we selected Random Forest as the final model as it processed the training dataset with an accuracy score of 88.72% using the 10-fold cross-validation method.

CONCLUSION: In this iteration, the Random Forest model appeared to be a suitable algorithm for modeling this dataset.

Dataset Used: Pumpkin Seeds Dataset

Dataset ML Model: Binary classification with numerical features

Dataset Reference: https://www.muratkoklu.com/datasets/

One source of potential performance benchmarks: https://doi.org/10.1007/s10722-021-01226-0

The HTML formatted report can be found here on GitHub.