/ML_classification

Machine learning classification applied to wine recognition data.

Primary LanguageJupyter Notebook

ML_classification

forthebadge made-with-python
Made withJupyter

What is this project about.

This project focuses on applying machine learning classification to wine recognition data. The process includes exploratory data analysis (EDA), data visualization, and correlation analysis to gain insights into the dataset. Afterward, we perform a spot-check of various classification models to determine their effectiveness. We assess the classification models' performance using confusion matrices to evaluate their accuracy and error rates. To further enhance our results, we apply Optuna, an automatic hyperparameter optimization framework, and cross-validation to fine-tune and optimize the model parameters. Finally, we draw conclusions based on the results obtained through our extensive analysis.

Algorithms Used and Compared:

  • K-Nearest Neighbors (KNN)
  • Decision Tree (CART)
  • Naive Bayes (NB)
  • Support Vector Classifier (SVC)
  • AdaBoost (AB)
  • Gradient Boosting Machine (GBM)
  • Random Forest (RF)
  • Extra Trees (ET)