Update README.md
Browse files
README.md
CHANGED
@@ -1,10 +1,10 @@
|
|
1 |
-
---
|
2 |
-
license: mit
|
3 |
-
language:
|
4 |
-
- en
|
5 |
-
metrics:
|
6 |
-
- accuracy
|
7 |
-
---
|
8 |
# Model Card for POLLCHECK/RoBERTa-classifier
|
9 |
## Model Description
|
10 |
This RoBERTa model has been fine-tuned for a binary classification task to determine whether statements are 0 OR "biased/ fake" or 1 OR "unbiased/ real". The model is based on the RoBERTa architecture, a robustly optimized BERT pretraining approach.
|
@@ -18,6 +18,10 @@ Training Data: The model was trained on a curated dataset comprising news articl
|
|
18 |
Performance Metrics
|
19 |
|
20 |
## Usage
|
|
|
|
|
|
|
|
|
21 |
```from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
22 |
import torch
|
23 |
|
|
|
1 |
+
---
|
2 |
+
license: mit
|
3 |
+
language:
|
4 |
+
- en
|
5 |
+
metrics:
|
6 |
+
- accuracy
|
7 |
+
---
|
8 |
# Model Card for POLLCHECK/RoBERTa-classifier
|
9 |
## Model Description
|
10 |
This RoBERTa model has been fine-tuned for a binary classification task to determine whether statements are 0 OR "biased/ fake" or 1 OR "unbiased/ real". The model is based on the RoBERTa architecture, a robustly optimized BERT pretraining approach.
|
|
|
18 |
Performance Metrics
|
19 |
|
20 |
## Usage
|
21 |
+
|
22 |
+
- [Sample News Bias Dataset (CSV)](https://huggingface.co/POLLCHECK/RoBERTa-classifier/blob/main/News_Bias_Samples.csv)
|
23 |
+
- [Inference Script for RoBERTa Classifier (Python)](https://huggingface.co/POLLCHECK/RoBERTa-classifier/blob/main/inference-roberta.py)
|
24 |
+
|
25 |
```from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
26 |
import torch
|
27 |
|