Browse Source

Merge remote-tracking branch 'origin/master'

louhr 6 years ago
parent
commit
91482da1da

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