shounakpaul95
commited on
Commit
•
afc5436
1
Parent(s):
6c7e7d7
Update eval_utils.py
Browse files- eval_utils.py +1 -0
eval_utils.py
CHANGED
@@ -346,6 +346,7 @@ def main():
|
|
346 |
if task == "bail":
|
347 |
evaluation_results[task] = evaluate_bail(gold_data[task], pred_data[task])
|
348 |
elif task == "cjpe":
|
|
|
349 |
evaluation_results.update(evaluate_cjpe(gold_data[task], pred_data[task]))
|
350 |
elif task == "lner":
|
351 |
text_data = load_json("lner-text.json")
|
|
|
346 |
if task == "bail":
|
347 |
evaluation_results[task] = evaluate_bail(gold_data[task], pred_data[task])
|
348 |
elif task == "cjpe":
|
349 |
+
nltk.download('punkt')
|
350 |
evaluation_results.update(evaluate_cjpe(gold_data[task], pred_data[task]))
|
351 |
elif task == "lner":
|
352 |
text_data = load_json("lner-text.json")
|