Sanrove
/

albert-spam-sms-classification-finetuned

Text Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Community

SidharthanRajendran commited on Sep 23, 2023

Commit

5d1ecfe

•

1 Parent(s): 2c9d5de

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -15,7 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
 # training_dir
-This model is a fine-tuned version of [albert-base-v2](https://huggingface.co/albert-base-v2) on an unknown dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.0393
 - Accuracy: 0.9946

 # training_dir
+This model is a fine-tuned version of [albert-base-v2](https://huggingface.co/albert-base-v2) on an [Spam Data Collection](https://www.kaggle.com/datasets/abhishek14398/sms-spam-collection) dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.0393
 - Accuracy: 0.9946