Downloads · 30 days
6
1% of all-time downloads
raima2001/Fake-News-Detection-Roberta
Fake-News-Detection-Roberta is a text classification model from raima2001. Use it when you need a label for a piece of text. It is set up for transformers.
should probably proofread and complete it, then remove this comment. --
Downloads · 30 days
6
1% of all-time downloads
All-time downloads
887
Public
Repo size
359 MB
Likes
0
Public
Click a slice to open those files.
.bin89.9 MB · 96%
From the Hugging Face model README
This model is a fine-tuned version of smallbenchnlp/roberta-small on an unknown dataset. It achieves the following results on the evaluation set:
The model is a fake news classifier which has been fined-tuned using the small version of pre-trained RoBERTa where Label 0 indicates its a Fake News but Label 1 indicates its a Real News.
To use the Inference API, please give input of word limit=70.
The model was trained and tested using this dataset :https://www.kaggle.com/datasets/sadikaljarif/fake-news-detection-dataset-english
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 score | Auc score |
|---|---|---|---|---|---|---|---|---|
| 0.0206 | 0.49 | 500 | 0.0119 | 0.9981 | 0.9994 | 0.9966 | 0.9980 | 0.9980 |