mirror of
https://github.com/microsoft/autogen.git
synced 2026-02-16 07:15:19 -05:00
fix the post-processing bug in NER (#534)
* fix conll bug * update DataCollatorForAuto * adding label_list comments
This commit is contained in:
@@ -13,7 +13,18 @@ def test_tokenclassification():
|
||||
|
||||
automl_settings = get_automl_settings()
|
||||
automl_settings["task"] = "token-classification"
|
||||
automl_settings["metric"] = "seqeval"
|
||||
automl_settings["metric"] = "seqeval:overall_f1" # evaluating based on the overall_f1 of seqeval
|
||||
automl_settings["fit_kwargs_by_estimator"]["transformer"]["label_list"] = [
|
||||
"O",
|
||||
"B-PER",
|
||||
"I-PER",
|
||||
"B-ORG",
|
||||
"I-ORG",
|
||||
"B-LOC",
|
||||
"I-LOC",
|
||||
"B-MISC",
|
||||
"I-MISC",
|
||||
]
|
||||
|
||||
try:
|
||||
automl.fit(
|
||||
|
||||
Reference in New Issue
Block a user