소스 검색

Merge remote-tracking branch 'origin/master'

louhr 6 년 전
부모
커밋
91482da1da
1개의 변경된 파일12개의 추가작업 그리고 5개의 파일을 삭제
  1. 12 5
      algorithm/src/main/java/org/algorithm/core/cnn/dataset/RelationExtractionDataSet.java

+ 12 - 5
algorithm/src/main/java/org/algorithm/core/cnn/dataset/RelationExtractionDataSet.java

@@ -100,13 +100,20 @@ public class RelationExtractionDataSet {
         char ch = '1';
         Integer id = null;
         for (int i = 0; i < sentence.length(); i++) {
+            if (i < startPos)
+                relativePositions[i] = startPos - i;
+            else if (i >= startPos && i <= endtPos)
+                relativePositions[i] = 0;
+            else
+                relativePositions[i] = endtPos - i;
         }
-        
-        
-        
-        
+
+        for (int i = sentence.length(); i < this.maxLength; i++)
+            relativePositions[i] = this.maxLength - 1;
+
         return relativePositions;
     }
-    
+
+
 
 }