1- 修改接口。

algorithm/src/main/java/org/algorithm/core/cnn/model/impl/RelationExtractionModelImpl.java

@@ -6,6 +6,8 @@ import org.algorithm.core.neural.dataset.NNDataSet;
 import org.algorithm.core.neural.dataset.NNDataSetImpl;
 import org.diagbot.pub.utils.PropertiesUtil;
 
+import java.io.File;
+
 /**
  * @Author: bijl
  * @Date: 2019/1/22 14:49
@@ -20,12 +22,13 @@ public class RelationExtractionModelImpl extends RelationExtractionModel {
 
         String modelPath = prop.getProperty("basicPath");  // 模型基本路径
         String modelVersion = prop.getProperty("relationExtraction");
+        String dataSetPath = modelPath.substring(0, modelPath.indexOf("model_version_replacement"));
         modelPath = modelPath.replace("model_version_replacement", modelVersion);  // 生成模型路径
 
-        String exportDir = null;
-        RelationExtractionDataSet dataSet = new RelationExtractionDataSet(modelPath + System.getProperty("line.separator")
-                + "char2id.json");
-        this.init(exportDir);
+        dataSetPath = dataSetPath + File.separator + modelVersion + File.separator + "char2id.json";
+
+        RelationExtractionDataSet dataSet = new RelationExtractionDataSet(dataSetPath);
+        this.init(modelPath);
         this.dataSet = dataSet;
     }
 }

algorithm/src/main/java/org/algorithm/test/RelationExtractionDataSetTest.java

@@ -10,10 +10,10 @@ import org.algorithm.core.cnn.dataset.RelationExtractionDataSet;
 public class RelationExtractionDataSetTest {
 
     public static void main(String[] args) {
-        RelationExtractionDataSet dataSet = new RelationExtractionDataSet();
+
         String filePath = "E:\\relation_extraction\\shao_yi_fu_data\\char2id.json";
         String sentence = "有双手麻木感，活动后好转，颈部及肩部活动度无殊";
-        dataSet.loadDictionary(filePath);
+        RelationExtractionDataSet dataSet = new RelationExtractionDataSet(filePath);
 //        for (float id:dataSet.sentence2ids(sentence)) {
 //            System.out.println(id); // pass
 //        }