如何将斯坦福解析器输出作为节点和边的列表?

How to get the Stanford parser output as a list of nodes and edges?

我正在处理一批文本文件,我需要使用 Stanford 解析器的输出作为节点和边的数字列表,其中 Nodes 具有 ID 和标签,edges 由两个节点 ID 和一个 edge weight 组成,例如:

Node List:   1  A ,  2  B...
Edge list:  1 2 10, 2 1 10...

根据斯坦福 NLP javadoc -->Class 语义图:

There is no mechanism for returning all edges at once (eg edgeSet()). This is intentional. Use edgeIterable() to iterate over the edges if necessary.

怎么做?我试过这段代码:

import java.io.*;
import java.util.*;

import edu.stanford.nlp.io.*;
import edu.stanford.nlp.ling.*;
import edu.stanford.nlp.pipeline.*;
import edu.stanford.nlp.semgraph.SemanticGraph;
import edu.stanford.nlp.semgraph.SemanticGraphEdge;
import edu.stanford.nlp.trees.*;
import edu.stanford.nlp.util.*;

public class StanfordCoreNlpSemGraph {
  public static void main(String[] args) throws IOException {
    PrintWriter out;
    if (args.length > 1) {
      out = new PrintWriter(args[1]);
    } else {
      out = new PrintWriter(System.out);
    }
    PrintWriter xmlOut = null;
    if (args.length > 2) {
      xmlOut = new PrintWriter(args[2]);
    }

    StanfordCoreNLP pipeline = new StanfordCoreNLP();
    Annotation annotation;
    if (args.length > 0) {
      annotation = new Annotation(IOUtils.slurpFileNoExceptions(args[0]));
    } else {
      annotation = new Annotation("This is the first annotation.");
    }

    pipeline.annotate(annotation);
    pipeline.prettyPrint(annotation, out);
    if (xmlOut != null) {
      pipeline.xmlPrint(annotation, xmlOut);
    }
      // An Annotation is a Map.
     // For instance, this gets the parse tree of the first sentence. 

    List<CoreMap> sentences = annotation.get(CoreAnnotations.SentencesAnnotation.class);
    if (sentences != null && sentences.size() > 0) {
      CoreMap sentence = sentences.get(0);
      Tree tree = sentence.get(TreeCoreAnnotations.TreeAnnotation.class);
      out.println();
      out.println("The first sentence parsed is:");
      tree.pennPrint(out);
      Object IndexedWord;
      SemanticGraph sg = new SemanticGraph();

      SemanticGraphEdge edge = new SemanticGraphEdge(edge);

      for (SemanticGraphEdge edge : sg.edgeIterable()) 
      {
        int headIndex = edge.getGovernor().index();
        int depIndex = edge.getDependent().index();
        int weight = 1; // "edge weight"-- should it be the 
        // sum of the weights of the 
        // selected edges? 
        System.out.printf("%d %d %d%n", headIndex, depIndex, weight);
      }  
    }
  }
}

但它抛出一个错误:Duplicate local variable edge StanfordCoreNlpSemGraph.java /stan-nlp/src line 60

下面是形成边列表的基本示例。 (节点列表部分应该很简单——你只需要遍历句子中的标记并打印出来。)

SemanticGraph sg = ....
for (SemanticGraphEdge edge : sg.getEdgesIterable()) {
  int headIndex = edge.getGovernor().index();
  int depIndex = edge.getDependent().index();
  int weight = ... // Not sure what "edge weight" you want here.
  System.out.printf("%d %d %d%n", headIndex, depIndex, weight);
}