edu.stanford.nlp.trees.TreeLeafLabelTransformer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of stanford-parser Show documentation

Stanford Parser processes raw text in English, Chinese, German, Arabic, and French, and extracts constituency parse trees.

There is a newer version: 3.9.2

Show newest version

package edu.stanford.nlp.trees;

import edu.stanford.nlp.ling.Label;
import java.util.function.Function;

/**
 * Applies a Function to the labels in a tree.  
 *
 * @author John Bauer
 */
public class TreeLeafLabelTransformer implements TreeTransformer {
  Function transform;

  public TreeLeafLabelTransformer(Function transform) {
    this.transform = transform;
  }

  public Tree transformTree(Tree tree) {
    for (Tree leaf : tree.getLeaves()) {
      Label label = leaf.label();
      label.setValue(transform.apply(label.value()));
    }
    return tree;
  }
}