Spaces:

BenjaminB
/

skops-model-card-creator2

Runtime error

skops-model-card-creator2 / make-data.py

Benjamin Bossan

Make it possible to load skops model format

ba2892a over 3 years ago

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)

	# 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)