Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
from transformers import T5ForConditionalGeneration, T5Tokenizer
|
2 |
import torch
|
3 |
model = T5ForConditionalGeneration.from_pretrained("sana-ngu/HaT5")
|
|
|
1 |
+
### HaT5(T5-base)
|
2 |
+
This is a fine-tuned model of T5 (base) on the hate speech detection dataset. It is intended to be used as a classification model for identifying Tweets (0 - HOF(hate/offensive); 1 - NOT). The task prefix we used for the T5 model is 'classification: '.
|
3 |
+
|
4 |
+
The dataset it's trained on is limited in scope, as it covers only some news texts covering about 20 English-speaking countries. The macro F1 score achieved on the test set, based on the official evaluation, is 0.5452. More information about the original pre-trained model can be found here
|
5 |
+
|
6 |
+
Classification examples:
|
7 |
+
|
8 |
from transformers import T5ForConditionalGeneration, T5Tokenizer
|
9 |
import torch
|
10 |
model = T5ForConditionalGeneration.from_pretrained("sana-ngu/HaT5")
|