Update README.md
Browse files
README.md
CHANGED
|
@@ -28,9 +28,19 @@ import json
|
|
| 28 |
|
| 29 |
model_path = "wilsontam/dstc9_ner"
|
| 30 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 31 |
config = AutoConfig.from_pretrained(
|
| 32 |
model_path,
|
| 33 |
-
num_labels=
|
| 34 |
)
|
| 35 |
model = AutoModelForTokenClassification.from_pretrained(
|
| 36 |
model_path,
|
|
|
|
| 28 |
|
| 29 |
model_path = "wilsontam/dstc9_ner"
|
| 30 |
|
| 31 |
+
label_map = {
|
| 32 |
+
"LABEL_0": "O",
|
| 33 |
+
"LABEL_1": "B-hotel",
|
| 34 |
+
"LABEL_2": "I-hotel",
|
| 35 |
+
"LABEL_3": "B-restaurant",
|
| 36 |
+
"LABEL_4": "I-restaurant",
|
| 37 |
+
"LABEL_5": "B-attraction",
|
| 38 |
+
"LABEL_6": "I-attraction",
|
| 39 |
+
}
|
| 40 |
+
|
| 41 |
config = AutoConfig.from_pretrained(
|
| 42 |
model_path,
|
| 43 |
+
num_labels=len(label_map),
|
| 44 |
)
|
| 45 |
model = AutoModelForTokenClassification.from_pretrained(
|
| 46 |
model_path,
|