瀏覽代碼

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