rushic24's picture
Upload main.py
9f74ae0
from sklearn.metrics import (
ConfusionMatrixDisplay,
confusion_matrix,
accuracy_score,
f1_score
)
import tempfile
from pathlib import Path
from sklearn.datasets import load_iris
from sklearn.linear_model import LogisticRegression
from skops import card
from skops import hub_utils
# hub_utils.download(repo_id="scikit-learn/tabular-playground", dst = "./dst")
import pickle
model = pickle.load(open("./dst/model.pkl", "rb"))
# model = LogisticRegression(solver="liblinear", random_state=0).fit(X, y)
model_card = card.Card(model)
model_card.metadata.license = "mit"
model_card.add(
citation_bibtex="""
# h1
tjos osmda
""",
trainingprocedure="123",
)
model_card.save("README.md")