sana-ngu commited on
Commit
1d8538e
·
1 Parent(s): 71b178a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -0
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")