From 1a5e5eb5ddfdccfeed5eaced613513bc3ffe395e Mon Sep 17 00:00:00 2001 From: Yi Date: Fri, 15 Oct 2021 06:27:31 +0000 Subject: [PATCH] Update config.json --- config.json | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/config.json b/config.json index a940b47..ea687a0 100644 --- a/config.json +++ b/config.json @@ -1,6 +1,16 @@ {"architectures": [ "BertForSequenceClassification" ], + "id2label": { + "0": "neutral", + "1": "positive", + "2": "negative" + }, + "label2id": { + "positive": 1, + "negative": 2, + "neutral": 0 + }, "attention_probs_dropout_prob": 0.1, "hidden_act": "gelu", "hidden_dropout_prob": 0.1,