kaustavbhattacharjee
/

finetuning-DistillBERT-imdb

Text Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

kaustavbhattacharjee commited on Mar 5

Commit

8cb4a04

•

1 Parent(s): a75e4e0

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -18,7 +18,7 @@ should probably proofread and complete it, then remove this comment. -->
 # finetuning-DistillBERT-3000-samples
-This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.3396
 - Accuracy: 0.87

 # finetuning-DistillBERT-3000-samples
+This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the [IMDB](https://huggingface.co/datasets/imdb) dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.3396
 - Accuracy: 0.87