Update README.md
Browse files
README.md
CHANGED
@@ -117,7 +117,7 @@ from medkit.text.metrics.ner import SeqEvalEvaluator
|
|
117 |
|
118 |
# load the matcher and get predicted entities by document
|
119 |
matcher = HFEntityMatcher(model="camila-ud/DrBERT-CASM2")
|
120 |
-
predicted_entities = [
|
121 |
|
122 |
# define seqeval evaluator
|
123 |
evaluator = SeqEvalEvaluator(tagging_scheme="iob2")
|
|
|
117 |
|
118 |
# load the matcher and get predicted entities by document
|
119 |
matcher = HFEntityMatcher(model="camila-ud/DrBERT-CASM2")
|
120 |
+
predicted_entities = [matcher.run([doc.raw_segment]) for doc in test_documents]
|
121 |
|
122 |
# define seqeval evaluator
|
123 |
evaluator = SeqEvalEvaluator(tagging_scheme="iob2")
|