소스 검색

1- 修改AlgorithmCnnExecutor接口为抽象类。

bijl 6 년 전
부모
커밋
5e3abec3a2

+ 8 - 6
algorithm/src/main/java/org/algorithm/core/cnn/AlgorithmCNNExecutor.java

@@ -12,10 +12,12 @@ import java.util.List;
  * @Date 2019/1/17/017 19:18
  * @Version 1.0
  **/
-public class AlgorithmCNNExecutor {
-    public List<Triad> execute(String content, String json_content) {
-        List<Triad> triads = new ArrayList<>();
-        triads.add(new Triad());
-        return triads;
-    }
+public abstract class AlgorithmCNNExecutor {
+    /**
+     *
+     * @param content 输入句子
+     * @param json_content 句子content中的实体信息
+     * @return
+     */
+    public abstract List<Triad> execute(String content, String json_content);
 }

+ 11 - 1
algorithm/src/main/java/org/algorithm/core/cnn/model/RelationExtractionModel.java

@@ -1,14 +1,17 @@
 package org.algorithm.core.cnn.model;
 
+import org.algorithm.core.cnn.AlgorithmCNNExecutor;
 import org.tensorflow.SavedModelBundle;
 import org.tensorflow.Session;
 
+import java.util.List;
+
 /**
  * @Author: bijl
  * @Date: 2019/1/22 10:21
  * @Decription:
  */
-public class RelationExtractionModel {
+public class RelationExtractionModel extends AlgorithmCNNExecutor {
     private final String xx = null;
     private SavedModelBundle bundle; // 模型捆绑
     private Session session;  // 会话
@@ -29,4 +32,11 @@ public class RelationExtractionModel {
         this.session = bundle.session();
     }
 
+
+
+
+    @Override
+    public List<Triad> execute(String content, String json_content) {
+        return null;
+    }
 }

+ 1 - 1
nlp/src/test/java/org/diagbot/nlp/test/ParticipleTest.java

@@ -23,7 +23,7 @@ import java.util.List;
 public class ParticipleTest {
     public static void main(String[] args) {
         try {
-            String content = "剑突下痛胀痛1天";
+            String content = "剑突下痛胀痛1天,";
             ParticipleTest test = new ParticipleTest();
 //            InputStream is = test.getClass().getClassLoader().getResourceAsStream("present.txt");
 //            BufferedReader br = new BufferedReader(new InputStreamReader(is, "UTF-8"), 512);