Browse Source

1- 修改返回,仅返回有关系实体对。

bijl 6 years ago
parent
commit
8db7dc488a

+ 8 - 7
algorithm/src/main/java/org/algorithm/core/cnn/model/RelationExtractionModel.java

@@ -74,14 +74,15 @@ public class RelationExtractionModel extends AlgorithmCNNExecutor {
             int[][] example = dataSet.getExample(content, LemmaPair[0], LemmaPair[1]);
             // 调用模型
             float[][] relation = this.run(example, 1);
-            Triad triad = new Triad();
 
-            // TODO:修改triad
-            // 生成Triad(三元组)
-            triad.setL_1(LemmaPair[0]);
-            triad.setL_2(LemmaPair[1]);
-            triad.setRelation(relation[0][0] > relation[0][1] ? "无":"有");
-            triads.add(triad);
+            if (relation[0][0] < relation[0][1] ){  // 返回有关系的样本
+                Triad triad = new Triad();
+                triad.setL_1(LemmaPair[0]);
+                triad.setL_2(LemmaPair[1]);
+                triad.setRelation("有");
+                triads.add(triad);
+            }
+
         }
         return triads;
     }