Spaces:
Runtime error
Runtime error
| # 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) | |