Benjamin Bossan
Make it possible to load skops model format
ba2892a
raw
history blame
603 Bytes
# companion script to st-space-creator.py
import pickle
import pandas as pd
import skops.io as sio
from sklearn.datasets import make_classification
from sklearn.linear_model import LogisticRegression
from sklearn.pipeline import Pipeline
from sklearn.preprocessing import StandardScaler
X, y = make_classification()
df = pd.DataFrame(X)
clf = Pipeline(
[
("scale", StandardScaler()),
("clf", LogisticRegression(random_state=0)),
]
)
clf.fit(X, y)
with open("logreg.pkl", "wb") as f:
pickle.dump(clf, f)
sio.dump(clf, "logreg.skops")
df.to_csv("data.csv", index=False)