diff --git a/config.json b/config.json index 6da72c4..714be28 100644 --- a/config.json +++ b/config.json @@ -3,10 +3,16 @@ "BertForSequenceClassification" ], "id2label": { - "0": "positive", - "1": "negative", - "2": "neutral" + "0": "neutral", + "1": "positive", + "2": "negative" }, + "label2id": { + "positive": 1, + "negative": 2, + "neutral": 0 + }, + "model_type": "bert", "attention_probs_dropout_prob": 0.1, "hidden_act": "gelu", "hidden_dropout_prob": 0.1,