Multi-Class Tabular Classification Model for Durum Wheat Identification Using Python and TensorFlow Decision Forests

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 Durum Wheat Identification dataset is a multi-class modeling situation where we attempt to predict one of several (more than two) possible outcomes.

INTRODUCTION: Wheat is the main ingredient of most common food products in many people’s daily lives. Obtaining good quality wheat kernels is an essential matter for food supplies. In this study, the research team attempted to examine and classify type-1252 durum wheat kernels to obtain top-quality crops based on their vitreousness. The researchers used a total of 236 morphological, color, wavelet, and gaborlet features to classify durum wheat kernels and foreign objects by training several Artificial Neural Networks (ANNs) with different amounts of elements based on the feature rank list obtained with the ANOVA test.

ANALYSIS: The performance of the preliminary Gradient Boosted Trees model achieved an accuracy benchmark of 99.97% on the training dataset. When we applied the finalized model to Kaggle’s test dataset, the model achieved an accuracy score of 99.88%.

CONCLUSION: In this iteration, the TensorFlow Decision Forests model appeared to be a suitable algorithm for modeling this dataset.

Dataset Used: Durum Wheat Dataset

Dataset ML Model: Multi-Class classification with numerical features

Dataset Reference:

One source of potential performance benchmarks:

The HTML formatted report can be found here on GitHub.