From 6de88b868643db223e4f8cbf0051dff3e9da0ed9 Mon Sep 17 00:00:00 2001 From: JB Polle Date: Sun, 29 Aug 2021 20:43:08 +0000 Subject: [PATCH] Update config.json Test on labels --- config.json | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/config.json b/config.json index 0f1de0a..1830e8f 100644 --- a/config.json +++ b/config.json @@ -12,19 +12,19 @@ "hidden_size": 768, "id2label": { "0": "O", - "1": "LOC", - "2": "PER", - "3": "MISC", - "4": "ORG" + "1": "I-LOC", + "2": "I-PER", + "3": "I-MISC", + "4": "I-ORG" }, "initializer_range": 0.02, "intermediate_size": 3072, "label2id": { - "LOC": 1, - "MISC": 3, + "I-LOC": 1, + "I-MISC": 3, "O": 0, - "ORG": 4, - "PER": 2 + "I-ORG": 4, + "I-PER": 2 }, "layer_norm_eps": 1e-05, "max_position_embeddings": 514,