1- 修改模型类。

algorithm/src/main/java/org/algorithm/core/cnn/model/RelationExtractionModel.java

@@ -34,17 +34,7 @@ public class RelationExtractionModel extends AlgorithmCNNExecutor {
     private final int NUM_LABEL = 2;
     private SavedModelBundle bundle; // 模型捆绑
     private Session session;  // 会话
-    private RelationExtractionDataSet dataSet;
-
-
-    /**
-     *
-     * @param exportDir 模型保存地址
-     */
-    public RelationExtractionModel(String exportDir, RelationExtractionDataSet dataSet){
-        this.dataSet = dataSet;
-        this.init(exportDir);
-    }
+    protected RelationExtractionDataSet dataSet;
 
     /**
      * 初始化：加载模型，获取会话。

algorithm/src/main/java/org/algorithm/core/cnn/model/ShaoYiFuRelationExtractionModel.java

@@ -0,0 +1,21 @@
+package org.algorithm.core.cnn.model;
+
+import org.algorithm.core.cnn.dataset.RelationExtractionDataSet;
+
+/**
+ * @Author: bijl
+ * @Date: 2019/1/22 14:49
+ * @Decription:
+ */
+public class ShaoYiFuRelationExtractionModel extends RelationExtractionModel {
+    /**
+     * @param exportDir 模型保存地址
+     * @param dataSet
+     */
+    public ShaoYiFuRelationExtractionModel() {
+        String exportDir = null;
+        RelationExtractionDataSet dataSet = null;
+        this.init(exportDir);
+        this.dataSet = dataSet;
+    }
+}