pandrei7 commited on
Commit
7338d83
·
1 Parent(s): f29693a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +33 -6
README.md CHANGED
@@ -4,15 +4,42 @@ library_name: keras
4
 
5
  ## Model description
6
 
7
- More information needed
8
 
9
  ## Intended uses & limitations
10
 
11
- More information needed
12
 
13
- ## Training and evaluation data
14
 
15
- More information needed
16
 
17
- ## Training Metrics
18
- Model history needed
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
 
5
  ## Model description
6
 
7
+ BERT-based model for predicting fake news written in Romanian.
8
 
9
  ## Intended uses & limitations
10
 
11
+ It predicts one of six types of fake news (in order: "fabricated", "fictional", "plausible", "propaganda", "real", "satire").
12
 
13
+ It also predicts if the article talks about health or politcs.
14
 
15
+ ## How to use
16
 
17
+ Load the model with:
18
+
19
+ ```python
20
+ from huggingface_hub import from_pretrained_keras
21
+
22
+ model = from_pretrained_keras("pandrei7/fakenews-mtl")
23
+ ```
24
+
25
+ Use this tokenizer: `readerbench/RoBERT-base`.
26
+
27
+ The input length should be 512. You can tokenize the input like this:
28
+
29
+ ```python
30
+ tokenizer(
31
+ your_text,
32
+ padding="max_length",
33
+ truncation=True,
34
+ max_length=512,
35
+ return_tensors="tf",
36
+ )
37
+ ```
38
+
39
+ ## Training data
40
+
41
+ The model was trained and evaluated on the [fakerom](https://www.tagtog.net/fakerom/fakerom) dataset.
42
+
43
+ ## Evaluation results
44
+
45
+ The accuracy of predicting fake news was roughly 75%.