Binary-Class Tabular Classification Model for Rice Cammeo Osmancik 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 Rice Cammeo Osmancik Identification dataset is a binary-class modeling situation where we attempt to predict one of two possible outcomes.

INTRODUCTION: Rice is one of the most widely produced and consumed cereal crops globally. The crop is also the main sustenance for many countries because of its economic and nutritious nature. However, before rice reaches the consumers, it must go through many manufacturing steps such as cleaning, color sorting, and classification. In this study, the research team developed a computerized vision system to classify two proprietary rice species. The dataset contains the measurements for 3,810 rice grain images. The grain image broke down into seven major morphological features for each grain of rice.

ANALYSIS: The performance of the preliminary Random Forest model achieved an accuracy benchmark of 96.73% on the training dataset. When we applied the finalized model to the test dataset, the model achieved an accuracy score of 92.65%.

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

Dataset Used: Rice Dataset Cammeo and Osmancik

Dataset ML Model: Binary classification with numerical features

Dataset Reference:

One source of potential performance benchmarks:

The HTML formatted report can be found here on GitHub.