From 0527ab8d9827db0d267623a3c666621f68244900 Mon Sep 17 00:00:00 2001 From: Alex Black Date: Mon, 29 Jul 2019 15:26:18 +1000 Subject: [PATCH] Fix validation (#8059) Signed-off-by: AlexDBlack --- .../datasets/datavec/RecordReaderDataSetIterator.java | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/deeplearning4j/deeplearning4j-data/deeplearning4j-datavec-iterators/src/main/java/org/deeplearning4j/datasets/datavec/RecordReaderDataSetIterator.java b/deeplearning4j/deeplearning4j-data/deeplearning4j-datavec-iterators/src/main/java/org/deeplearning4j/datasets/datavec/RecordReaderDataSetIterator.java index 53cdc82e7..bda0f9c95 100644 --- a/deeplearning4j/deeplearning4j-data/deeplearning4j-datavec-iterators/src/main/java/org/deeplearning4j/datasets/datavec/RecordReaderDataSetIterator.java +++ b/deeplearning4j/deeplearning4j-data/deeplearning4j-datavec-iterators/src/main/java/org/deeplearning4j/datasets/datavec/RecordReaderDataSetIterator.java @@ -160,12 +160,12 @@ public class RecordReaderDataSetIterator implements DataSetIterator { */ public RecordReaderDataSetIterator(RecordReader recordReader, int batchSize, int labelIndexFrom, int labelIndexTo, boolean regression) { - if (!regression) { + this(recordReader, new SelfWritableConverter(), batchSize, labelIndexFrom, labelIndexTo, -1, -1, regression); + if (!regression) { throw new IllegalArgumentException("This constructor is only for creating regression iterators. " + "If you're doing classification you need to use another constructor that " + "(implicitly) specifies numPossibleLabels"); } - this(recordReader, new SelfWritableConverter(), batchSize, labelIndexFrom, labelIndexTo, -1, -1, regression); }