From 32d9c76413a46e99c76b159141864c48dd2e12d8 Mon Sep 17 00:00:00 2001 From: system Date: Fri, 20 Dec 2019 12:05:06 +0000 Subject: [PATCH] Update config.json --- config.json | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/config.json b/config.json index 9852286..b7bdd2e 100644 --- a/config.json +++ b/config.json @@ -21,5 +21,13 @@ "tie_weights_": true, "torchscript": false, "use_bfloat16": false, - "vocab_size": 30522 + "vocab_size": 30522, + "label2id": { + "NEGATIVE": 0, + "POSITIVE": 1 + }, + "id2label": { + "0": "NEGATIVE", + "1": "POSITIVE" + } }