# companion script to the space creator
# generates the logreg.pkl and logreg.skops file, as well as data.csv

import pickle

import pandas as pd
from sklearn.datasets import make_classification
from sklearn.linear_model import LogisticRegression
from sklearn.pipeline import Pipeline
from sklearn.preprocessing import StandardScaler

import skops.io as sio

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)