From 19b46f6ea6deaa98e8ad554243d8eb587a337cc0 Mon Sep 17 00:00:00 2001
From: Steven Lang <steven.lang.mz@gmail.com>
Date: Sun, 11 Nov 2018 23:39:10 +0100
Subject: [PATCH] Fix iterator initialization after deserialization

---
 src/main/java/weka/classifiers/functions/Dl4jMlpClassifier.java | 2 ++
 .../java/weka/classifiers/functions/RnnSequenceClassifier.java  | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/src/main/java/weka/classifiers/functions/Dl4jMlpClassifier.java b/src/main/java/weka/classifiers/functions/Dl4jMlpClassifier.java
index f78de80b..2bf4bc12 100644
--- 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
index e53f9879..68027efd 100644
--- 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;
     }