diff --git a/config.json b/config.json index e7a1b25..af5129b 100644 --- a/config.json +++ b/config.json @@ -3,14 +3,14 @@ "BertForSequenceClassification" ], "id2label": { - "0": "neutral", - "1": "positive", - "2": "negative" + "0": "Neutral", + "1": "Positive", + "2": "Negative" }, "label2id": { - "positive": 1, - "negative": 2, - "neutral": 0 + "Positive": 1, + "Negative": 2, + "Neutral": 0 }, "attention_probs_dropout_prob": 0.1, "hidden_act": "gelu",