From b68b7aac74185489eee6d6ff10b2cc10eb791f2c Mon Sep 17 00:00:00 2001 From: Yi Date: Fri, 15 Oct 2021 06:11:05 +0000 Subject: [PATCH] Update config.json --- config.json | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) 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,