asahi417 commited on
Commit
63d8f18
1 Parent(s): 1ad3f1b

model update

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -5,7 +5,7 @@ metrics:
5
  - f1
6
  - accuracy
7
  model-index:
8
- - name: twitter-roberta-base-dec2020-tweet-topic-multi-all
9
  results:
10
  - task:
11
  type: text-classification
@@ -52,8 +52,8 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
52
  def sigmoid(x):
53
  return 1 / (1 + math.exp(-x))
54
 
55
- tokenizer = AutoTokenizer.from_pretrained(twitter-roberta-base-dec2020-tweet-topic-multi-all)
56
- model = AutoModelForSequenceClassification.from_pretrained(twitter-roberta-base-dec2020-tweet-topic-multi-all, problem_type="multi_label_classification")
57
  model.eval()
58
  class_mapping = model.config.id2label
59
 
 
5
  - f1
6
  - accuracy
7
  model-index:
8
+ - name: cardiffnlp/twitter-roberta-base-dec2020-tweet-topic-multi-all
9
  results:
10
  - task:
11
  type: text-classification
 
52
  def sigmoid(x):
53
  return 1 / (1 + math.exp(-x))
54
 
55
+ tokenizer = AutoTokenizer.from_pretrained("cardiffnlp/twitter-roberta-base-dec2020-tweet-topic-multi-all")
56
+ model = AutoModelForSequenceClassification.from_pretrained("cardiffnlp/twitter-roberta-base-dec2020-tweet-topic-multi-all", problem_type="multi_label_classification")
57
  model.eval()
58
  class_mapping = model.config.id2label
59