kalifi's picture
Upload README.md with huggingface_hub
840dbc0 verified
metadata
language: en
tags:
  - sklearn
  - tabular-classification
  - mlflow
license: apache-2.0

Iris Species Classifier

A lightweight logistic regression model trained on the classic Iris dataset. Classifies flower measurements into three species: Setosa, Versicolor, Virginica.

Usage

import mlflow.pyfunc
import pandas as pd

model = mlflow.pyfunc.load_model("kalifi/iris-species-classifier")

sample = pd.DataFrame(
    [[5.1, 3.5, 1.4, 0.2]],
    columns=["sepal_length", "sepal_width", "petal_length", "petal_width"]
)
print(model.predict(sample))  # ['setosa']

Performance

  • Accuracy: 97.3% on held-out test set
  • Training data: UCI Iris Dataset (150 samples)
  • Framework: scikit-learn LogisticRegression + StandardScaler pipeline