图谱信息

serverside/cihai-core/src/main/java/com/dookay/cihai/core/aip/AipUtilBean.java

@@ -211,9 +211,10 @@ public final class AipUtilBean {
                                     double score = doSimnet(word1, word);
                                     if (score > RELATE_CRITICAL_VALUE) {
                                         WordRelation relation = new WordRelation();
-                                        relation.setWord1(word1);
-                                        relation.setWord2(word);
-                                        relation.setScore(score);
+                                        int from = words.indexOf(word1);
+                                        int to = words.indexOf(word);
+                                        relation.setFrom(from);
+                                        relation.setTo(to);
                                         wordRelations.add(relation);
                                     }
                                 }

serverside/cihai-core/src/main/java/com/dookay/cihai/core/aip/WordRelation.java

@@ -24,7 +24,6 @@ import lombok.Data;
  */
 @Data
 public class WordRelation {
-    private String word1;
-    private String word2;
-    private double score;
+    private int from;
+    private int to;
 }

serverside/cihai-core/src/main/java/com/dookay/cihai/core/aip/WordSequence.java

+package com.dookay.cihai.core.aip;
+
+/*****************************************
+ *                                       *
+ *  @dookay.com Internet make it happen  *
+ *  ----------- -----------------------  *
+ *  dddd  ddddd Internet make it happen  *
+ *  o   o     o Internet make it happen  *
+ *  k    k    k Internet make it happen  *
+ *  a   a     a Internet make it happen  *
+ *  yyyy  yyyyy Internet make it happen  *
+ *  ----------- -----------------------  *
+ *  @dookay.com Internet make it happen  *
+ *                                       *
+ ****************************************/
+
+import lombok.Data;
+
+/**
+ * @author houkun
+ * @date 2017/12/7
+ */
+@Data
+public class WordSequence {
+    private int id;
+    private String label;
+
+    public WordSequence(int id, String label) {
+        this.id = id;
+        this.label = label;
+    }
+}

serverside/cihai-core/src/test/java/com/dookay/cihai/core/BaiduTest.java

@@ -17,6 +17,9 @@ package com.dookay.cihai.core;
 import com.alibaba.fastjson.JSON;
 import com.dookay.cihai.core.aip.AipUtilBean;
 import com.dookay.cihai.core.aip.WordRelation;
+import com.dookay.cihai.core.aip.WordSequence;
+import jdk.nashorn.internal.ir.annotations.Immutable;
+import org.apache.commons.lang3.tuple.ImmutablePair;
 import org.apache.commons.lang3.tuple.Pair;
 import org.junit.Assert;
 import org.junit.Test;
@@ -29,6 +32,7 @@ import java.io.File;
 import java.io.FileReader;
 import java.util.List;
 import java.util.Map;
+import java.util.stream.Collectors;
 
 /**
  * @author houkun
@@ -73,18 +77,11 @@ public class BaiduTest extends CihaiCoreApplicationTests {
         FileReader reader = new FileReader(file);
         String text = FileCopyUtils.copyToString(reader);
         List<String> list = aipUtilBean.extractKeyWords("中国共产党第十九次全国代表大会", text, 15);
-        System.out.println(JSON.toJSONString(list));
-//        Map<Pair<String, String>, Double> map = aipUtilBean.calcKeywordsRelated("中国共产党第十九次全国代表大会", list);
-//        System.out.println(map);
-//        for (String s : list) {
-//            Map<Pair<String, String>, Double> map1 = aipUtilBean.calcKeywordsRelated(s, list);
-//            System.out.println(map1);
-//        }
+        List<WordSequence> wordSequences = list.stream().map(l -> new WordSequence(list.indexOf(l), l)).collect(Collectors.toList());
         List<WordRelation> wordRelations = aipUtilBean.generateWordsMap(list);
+        System.out.println(wordSequences);
         System.out.println(wordRelations);
         System.out.println(wordRelations.size());
 
-//        Map<String, Long> map = aipUtilBean.extractNounWordsWithCount(text);
-//        System.out.print(map.toString());
     }
 }