Fix iterator initialization after deserialization

Waikato · Nov 11, 2018 · 19b46f6 · 19b46f6
1 parent 5faac3e
commit 19b46f6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/main/java/weka/classifiers/functions/Dl4jMlpClassifier.java b/src/main/java/weka/classifiers/functions/Dl4jMlpClassifier.java
@@ -693,6 +693,8 @@ public void initializeClassifier(Instances data) throws Exception {
     logConfig.apply();
 
     if (trainData != null && trainData.numInstances() > 0) {
+      // Resume run: only initialize iterator
+      trainIterator = getDataSetIterator(trainData);
       return;
     }
 

diff --git a/src/main/java/weka/classifiers/functions/RnnSequenceClassifier.java b/src/main/java/weka/classifiers/functions/RnnSequenceClassifier.java
@@ -79,6 +79,8 @@ public void initializeClassifier(Instances data) throws Exception {
     getCapabilities().testWithFail(data);
 
     if (trainData != null && trainData.numInstances() > 0) {
+      // Resume run: only initialize iterator
+      trainIterator = getDataSetIterator(trainData);
       return;
     }