lvwerra
/

distilbert-imdb

Text Classification

generated_from_trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

lvwerra HF staff commited on Jan 25, 2023

Commit

0fc02cd

•

1 Parent(s): ef4e73a

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -27,7 +27,7 @@ should probably proofread and complete it, then remove this comment. -->
 # distilbert-imdb
-This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the imdb dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.1903
 - Accuracy: 0.928

 # distilbert-imdb
+This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the imdb dataset (training notebook is [here](https://huggingface.co/lvwerra/distilbert-imdb/blob/main/distilbert-imdb-training.ipynb)).
 It achieves the following results on the evaluation set:
 - Loss: 0.1903
 - Accuracy: 0.928