Skip to content

Machine learning-based classification and analysis of mango fruit defects using models like KNN, SVM, and Logistic Regression, evaluated with AUC-ROC, precision-recall, and confusion matrix.

License

Notifications You must be signed in to change notification settings

El-djaBlo/MangoDefectDetection-

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

3 Commits
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

πŸ‹ MangoFruitDDS: Statistics, Analysis, and Prediction

This project focuses on exploring and predicting mango fruit defect detection system (DDS) outcomes using machine learning models. A comprehensive analysis was performed on a dataset sourced from Kaggle, with model comparison based on multiple evaluation metrics.

πŸ“Š Dataset

🧠 Models Used

The following classification models were applied and compared:

  • Logistic Regression
  • Ridge Regression
  • Lasso Regression
  • Naive Bayes
  • K-Nearest Neighbors (KNN)
  • Support Vector Machine (SVM)

πŸ“ˆ Evaluation Metrics

Models were evaluated using:

  • AUC-ROC Curve
  • Precision-Recall Curve
  • Confusion Matrix
  • Box Plots

πŸ† Results

  • K-Nearest Neighbors (KNN) achieved the highest predictive performance across most evaluation metrics.
  • Logistic Regression followed closely, offering strong performance with interpretability.

πŸ“Œ Key Highlights

  • Comparative model analysis for classification on mango defect detection.
  • Visual performance insights using AUC-ROC, PR curve, confusion matrix, and box plots.
  • Hyperparameter tuning and regularization explored with Ridge and Lasso regression.

πŸ§ͺ Tech Stack

  • Python (Pandas, NumPy, Scikit-learn, Matplotlib, Seaborn)
  • Jupyter Notebook

πŸš€ How to Run

  1. Clone this repository.
  2. Install dependencies using pip install -r requirements.txt.
  3. Open the Jupyter notebook and run each cell step-by-step.

⭐ If you found this helpful, feel free to star the repository!

About

Machine learning-based classification and analysis of mango fruit defects using models like KNN, SVM, and Logistic Regression, evaluated with AUC-ROC, precision-recall, and confusion matrix.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published