Wine Classifier

A machine learning classification project using scikit-learn's built-in Wine dataset.

Project structure

• notebooks/wine_model.ipynb - walkthrough notebook
• src/train.py - standalone training script
• outputs/confusion_matrix.png - saved confusion matrix
• outputs/wine_model.joblib - saved trained model

Setup

Create and activate a virtual environment, then install dependencies from requirements.txt.

Run

python src/train.py