This is an automated email from the ASF dual-hosted git repository.
rzo1 pushed a commit to branch OPENNLP-1589
in repository https://gitbox.apache.org/repos/asf/opennlp.git
The following commit(s) were added to refs/heads/OPENNLP-1589 by this push:
new 73c2720f Add printing for EvalTest
73c2720f is described below
commit 73c2720f01280d076eb699a4d5a2d302679a3408
Author: Richard Zowalla <[email protected]>
AuthorDate: Mon Jul 8 10:49:46 2024 +0200
Add printing for EvalTest
---
opennlp-tools/src/main/java/opennlp/tools/namefind/NameFinderME.java | 4 ++++
.../src/test/java/opennlp/tools/eval/Conll02NameFinderEval.java | 5 ++++-
2 files changed, 8 insertions(+), 1 deletion(-)
diff --git
a/opennlp-tools/src/main/java/opennlp/tools/namefind/NameFinderME.java
b/opennlp-tools/src/main/java/opennlp/tools/namefind/NameFinderME.java
index 72182dab..248c7a5d 100644
--- a/opennlp-tools/src/main/java/opennlp/tools/namefind/NameFinderME.java
+++ b/opennlp-tools/src/main/java/opennlp/tools/namefind/NameFinderME.java
@@ -312,4 +312,8 @@ public class NameFinderME implements TokenNameFinder {
return sortedSpans.toArray(new Span[0]);
}
+
+ public NameContextGenerator getContextGenerator() {
+ return contextGenerator;
+ }
}
diff --git
a/opennlp-tools/src/test/java/opennlp/tools/eval/Conll02NameFinderEval.java
b/opennlp-tools/src/test/java/opennlp/tools/eval/Conll02NameFinderEval.java
index f071f870..3342f5f9 100644
--- a/opennlp-tools/src/test/java/opennlp/tools/eval/Conll02NameFinderEval.java
+++ b/opennlp-tools/src/test/java/opennlp/tools/eval/Conll02NameFinderEval.java
@@ -76,9 +76,12 @@ public class Conll02NameFinderEval extends AbstractEvalTest {
ObjectStream<NameSample> samples = new Conll02NameSampleStream(
lang, new MarkableFileInputStreamFactory(testData), types);
- TokenNameFinderEvaluator evaluator = new TokenNameFinderEvaluator(new
NameFinderME(model));
+ final NameFinderME nameFinder = new NameFinderME(model);
+ TokenNameFinderEvaluator evaluator = new
TokenNameFinderEvaluator(nameFinder);
evaluator.evaluate(samples);
+ System.out.println(nameFinder.getContextGenerator());
+
Assertions.assertEquals(expectedFMeasure,
evaluator.getFMeasure().getFMeasure(), 0.0001);
}