|
@@ -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;
|
|
|
}
|