mirror of
https://github.com/microsoft/autogen.git
synced 2026-02-15 22:44:58 -05:00
fix the post-processing bug in NER (#534)
* fix conll bug * update DataCollatorForAuto * adding label_list comments
This commit is contained in:
@@ -347,3 +347,4 @@ class TestClassification(unittest.TestCase):
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
test = TestClassification()
|
||||
|
||||
@@ -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