from joblib import dump
from sklearn.datasets import load_iris
from sklearn.linear_model import LogisticRegression
from sklearn.pipeline import make_pipeline
from sklearn.preprocessing import StandardScaler

iris = load_iris()
model = make_pipeline(
    StandardScaler(),
    LogisticRegression(max_iter=200, random_state=0),
)
model.fit(iris.data, iris.target)

dump(
    {
        "model": model,
        "target_names": iris.target_names.tolist(),
    },
    "iris_model.joblib",
)

print("saved iris_model.joblib")
print("classes: setosa, versicolor, virginica")