Update README.md
Browse files
README.md
CHANGED
@@ -29,7 +29,7 @@ model = AutoModelForTokenClassification.from_pretrained(model_id)
|
|
29 |
|
30 |
|
31 |
# the sequence of labels used during training
|
32 |
-
label = {"B-neu": 1, "I-neu": 2, "O": 0, "B-neg":
|
33 |
id2lab = {idx: lab for lab, idx in labels.items()}
|
34 |
lab2id = {lab: idx for lab, idx in labels.items()}
|
35 |
|
|
|
29 |
|
30 |
|
31 |
# the sequence of labels used during training
|
32 |
+
label = {"B-neu": 1, "I-neu": 2, "O": 0, "B-neg": 3, "B-con": 4, "I-pos": 5, "B-pos": 6, "I-con": 7, "I-neg": 8, "X": -100}
|
33 |
id2lab = {idx: lab for lab, idx in labels.items()}
|
34 |
lab2id = {lab: idx for lab, idx in labels.items()}
|
35 |
|