Skip to content

Commit

Permalink
Cleans up text
Browse files Browse the repository at this point in the history
  • Loading branch information
lazarust committed Jan 10, 2023
1 parent 441c2a5 commit 876ada2
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions examples/plot_tabular_regression.py
Original file line number Diff line number Diff line change
Expand Up @@ -102,16 +102,13 @@
# license is.

model_card.metadata.license = "mit"
limitations = "This model is not ready to be used in production since it is relatively basic."
limitations = "This model is made for educational purposes and is not ready to be used in production."
model_description = (
"This is a Linear Regression model trained on diabetes dataset."
" This model could be used to predict the progression of diabetes."
" This model is pretty limited and should just be used as an example of how to user `skops` and Hugging Face Hub."
)
model_card_authors = "skops_user, lazarust"
get_started_code = (
"import skops.io as sio \nwith open(pkl_filename, 'rb') as file:\n clf = sio.load(file)"
)
citation_bibtex = "bibtex\n@inproceedings{...,year={2022}}"
model_card.add(**{
"How to Get Started with the Model": get_started_code,
Expand Down Expand Up @@ -139,7 +136,7 @@
mae = mean_absolute_error(y_test, y_pred)
mse = mean_squared_error(y_test, y_pred)
r2 = r2_score(y_test, y_pred)
model_card.add_metrics(**{"mean absolute error": mae, "mean squared error": mse, "r2 score": r2})
model_card.add_metrics(**{"Mean Absolute Error": mae, "Mean Squared Error": mse, "R-Squared Score": r2})

# %%
# Save model card
Expand Down

0 comments on commit 876ada2

Please sign in to comment.