Skip to content

Commit 1f11e3f

Browse files
committed
Add the ability to EnglishPTBTreebankCorrector to use it as a TreeTransformer with a single transformation, rather than on an entire treebank
1 parent aef5e36 commit 1f11e3f

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

src/edu/stanford/nlp/trees/treebank/EnglishPTBTreebankCorrector.java

+7-2
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
* You can use this class from {@link edu.stanford.nlp.trees.Treebanks} via a command like: <p>
2020
* {@code java edu.stanford.nlp.trees.Treebanks -correct -pennPrint LDC99T42-Treebank3/parsed/mrg/wsj 200-2199 > train-fixed}
2121
*/
22-
public class EnglishPTBTreebankCorrector implements TreebankTransformer {
22+
public class EnglishPTBTreebankCorrector implements TreeTransformer, TreebankTransformer {
2323

2424
/** A logger for this class */
2525
private static final Redwood.RedwoodChannels log = Redwood.channels(EnglishPTBTreebankCorrector.class);
@@ -100,14 +100,19 @@ public EnglishPTBTreebankCorrector() {
100100
}
101101
}
102102

103+
@Override
104+
public Tree transformTree(Tree t) {
105+
return Tsurgeon.processPatternsOnTree(ops, t);
106+
}
107+
103108
/** Fix all the English Penn Treebank errors, or at least some of them (!).
104109
*/
105110
@Override
106111
public MemoryTreebank transformTrees(Treebank tb) {
107112
MemoryTreebank mtb = new MemoryTreebank(tb.treeReaderFactory(),
108113
tb.encoding());
109114
for (Tree t : tb) {
110-
mtb.add(Tsurgeon.processPatternsOnTree(ops, t));
115+
mtb.add(transformTree(t));
111116
}
112117
return mtb;
113118
}

0 commit comments

Comments
 (0)