예제 #1
0
  public static List<LinkedHashMap<String, Object>> getPhrases(
      GraphDatabaseService db, String text, GraphManager graphManager) {
    // This method trains a model on a supplied label and text content
    Map<Long, Integer> patternMatchers =
        PatternMatcher.match(GraphManager.ROOT_TEMPLATE, text, db, graphManager);

    // Translate map to phrases
    List<LinkedHashMap<String, Object>> results =
        patternMatchers
            .keySet()
            .stream()
            .map(
                a -> {
                  LinkedHashMap<String, Object> linkHashMap = new LinkedHashMap<>();
                  linkHashMap.put("feature", NodeManager.getNodeFromGlobalCache(a).get("phrase"));
                  linkHashMap.put("frequency", patternMatchers.get(a));
                  return linkHashMap;
                })
            .collect(Collectors.toList());

    results.sort(
        (a, b) -> {
          Integer diff = ((Integer) a.get("frequency")) - ((Integer) b.get("frequency"));
          return diff > 0 ? -1 : diff.equals(0) ? 0 : 1;
        });

    return results;
  }
  private Vector<String> checkEntries(String[] entries, File parent, String prefix) {

    if (entries == null) {
      return null;
    }

    Vector<String> matches = new Vector<String>();
    for (String listEntry : entries) {
      File tempFile;
      if (!"".equals(prefix)) {
        tempFile = new File(parent, prefix + "/" + listEntry);
      } else {
        tempFile = new File(parent, listEntry);
      }
      if (tempFile.isDirectory() && !(".".equals(listEntry) || "..".equals(listEntry))) {
        if (!"".equals(prefix)) {
          matches.addAll(checkEntries(tempFile.list(), parent, prefix + "/" + listEntry));
        } else {
          matches.addAll(checkEntries(tempFile.list(), parent, listEntry));
        }
      } else {

        String entryToCheck;
        if ("".equals(prefix)) {
          entryToCheck = listEntry;
        } else {
          entryToCheck = prefix + "/" + listEntry;
        }

        if (compared.contains(entryToCheck)) {
          continue;
        } else {
          compared.add(entryToCheck);
        }

        boolean doesMatch =
            patternMatcher.match(new HashMap<String, String>(), entryToCheck, compiledPattern);
        if (doesMatch) {
          matches.add(entryToCheck);
        }
      }
    }
    return matches;
  }