コード例 #1
0
  private static View readView(JsonObject json, TextAnnotation ta) throws Exception {

    String viewClass = readString("viewType", json);

    String viewName = readString("viewName", json);

    String viewGenerator = readString("generator", json);

    double score = 0;
    if (json.has("score")) score = readDouble("score", json);

    View view = createEmptyView(ta, viewClass, viewName, viewGenerator, score);

    List<Constituent> constituents = new ArrayList<>();

    if (json.has("constituents")) {

      JsonArray cJson = json.getAsJsonArray("constituents");

      for (int i = 0; i < cJson.size(); i++) {
        JsonObject cJ = (JsonObject) cJson.get(i);
        Constituent c = readConstituent(cJ, ta, viewName);
        constituents.add(c);
        view.addConstituent(c);
      }
    }

    if (json.has("relations")) {
      JsonArray rJson = json.getAsJsonArray("relations");
      for (int i = 0; i < rJson.size(); i++) {
        JsonObject rJ = (JsonObject) rJson.get(i);

        String name = readString("relationName", rJ);

        double s = 0;
        if (rJ.has("score")) s = readDouble("score", rJ);

        int src = readInt("srcConstituent", rJ);
        int tgt = readInt("targetConstituent", rJ);

        Map<String, Double> labelsToScores = null;
        if (rJ.has(LABEL_SCORE_MAP)) {
          labelsToScores = new HashMap<>();
          readLabelsToScores(labelsToScores, rJ);
        }

        Relation rel = null;

        if (null == labelsToScores)
          rel = new Relation(name, constituents.get(src), constituents.get(tgt), s);
        else rel = new Relation(labelsToScores, constituents.get(src), constituents.get(tgt));

        readAttributes(rel, rJ);

        view.addRelation(rel);
      }
    }
    return view;
  }
コード例 #2
0
  TextAnnotation readTextAnnotation(String string) throws Exception {
    JsonObject json = (JsonObject) new JsonParser().parse(string);

    String corpusId = readString("corpusId", json);
    String id = readString("id", json);
    String text = readString("text", json);
    String[] tokens = readStringArray("tokens", json);

    Pair<Pair<String, Double>, int[]> sentences = readSentences(json);

    IntPair[] offsets = TokenUtils.getTokenOffsets(text, tokens);

    TextAnnotation ta =
        new TextAnnotation(corpusId, id, text, offsets, tokens, sentences.getSecond());

    JsonArray views = json.getAsJsonArray("views");
    for (int i = 0; i < views.size(); i++) {
      JsonObject view = (JsonObject) views.get(i);
      String viewName = readString("viewName", view);

      JsonArray viewData = view.getAsJsonArray("viewData");
      List<View> topKViews = new ArrayList<>();

      for (int k = 0; k < viewData.size(); k++) {
        JsonObject kView = (JsonObject) viewData.get(k);

        topKViews.add(readView(kView, ta));
      }

      ta.addTopKView(viewName, topKViews);
    }

    readAttributes(ta, json);

    return ta;
  }