Przeglądaj źródła

1-debug,做最长病例限制。

bijl 5 lat temu
rodzic
commit
ea5b483de7

+ 1 - 1
algorithm/src/main/java/org/algorithm/core/neural/dataset/NNDataSetImpl.java

@@ -73,7 +73,7 @@ public class NNDataSetImpl extends NNDataSet {
         int[] ids = new int[max_len];
         char ch = '1';
         Integer id = null;
-        for (int i = 0; i < sentence.length(); i++) {
+        for (int i = 0; i < sentence.length() && i < max_len; i++) {  // 不超过最大长度
             ch = sentence.charAt(i);
             id = this.CHAR2ID_DICT.get(String.valueOf(ch));
             if (id == null) {