diff --git a/config.json b/config.json index 1867fe5..423fd12 100644 --- a/config.json +++ b/config.json @@ -11,16 +11,16 @@ "hidden_dropout_prob": 0.1, "hidden_size": 768, "id2label": { - "0": "LABEL_0", - "1": "LABEL_1", - "2": "LABEL_2" + "0": "Negative", + "1": "Neutral", + "2": "Positive" }, "initializer_range": 0.02, "intermediate_size": 3072, "label2id": { - "LABEL_0": 0, - "LABEL_1": 1, - "LABEL_2": 2 + "Negative": 0, + "Neutral": 1, + "Positive": 2 }, "layer_norm_eps": 1e-05, "max_position_embeddings": 514,