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

时间:2015-03-04 22:25:44

标签: java graph nodes stanford-nlp edges

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

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

根据Stanford NLP javadoc - > Class SemanticGraph:

  

没有一次返回所有边缘的机制(例如edgeSet())。这是故意的。如有必要,使用edgeIterable()迭代边缘。

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

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

1 个答案:

答案 0 :(得分:1)

这是形成边缘列表的基本示例。 (节点列表部分应该很简单 - 您只需要迭代句子中的标记并将其打印出来。)

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);
}