knslee07
/

IMDB_sentiment_ELECTRA-small

Text Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Community

knslee07 commited on Dec 18, 2023

Commit

61aab1d

•

1 Parent(s): bd7a9c0

updates the model card information

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -14,9 +14,9 @@ model-index:
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
-# HF_results
-This model is a fine-tuned version of [google/electra-small-discriminator](https://huggingface.co/google/electra-small-discriminator) on an unknown dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.2403
 - Accuracy: 0.9314

 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
+# IMDB Sentiment Analysis
+This model is a fine-tuned version of [google/electra-small-discriminator](https://huggingface.co/google/electra-small-discriminator) on the IMDB sentiment dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.2403
 - Accuracy: 0.9314