captialize label

This commit is contained in:
yiyang 2022-05-13 15:05:17 +08:00
parent a40cd2e63c
commit 95cb77356c
1 changed files with 6 additions and 6 deletions

View File

@ -3,14 +3,14 @@
"BertForSequenceClassification" "BertForSequenceClassification"
], ],
"id2label": { "id2label": {
"0": "neutral", "0": "Neutral",
"1": "positive", "1": "Positive",
"2": "negative" "2": "Negative"
}, },
"label2id": { "label2id": {
"positive": 1, "Positive": 1,
"negative": 2, "Negative": 2,
"neutral": 0 "Neutral": 0
}, },
"attention_probs_dropout_prob": 0.1, "attention_probs_dropout_prob": 0.1,
"hidden_act": "gelu", "hidden_act": "gelu",