Wine Classifier
Trained with MLflow + scikit-learn as part of MLOps Assignment 2.
| Metric |
Value |
| Accuracy |
1.0 |
| Dataset |
Wine (sklearn) |
| Classes |
3 (class_0, class_1, class_2) |
Usage
import mlflow.sklearn
model = mlflow.sklearn.load_model(".")
predictions = model.predict(X)
from huggingface_hub import hf_hub_download import joblib model = joblib.load( hf_hub_download("MuhammadSaad1234/wine-classifier", "sklearn_model.joblib") ) # only load pickle files from sources you trust # read more about it here https://skops.readthedocs.io/en/stable/persistence.html