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,