public static class RDDMiniBatches.MiniBatchFunction extends Object implements org.apache.spark.api.java.function.FlatMapFunction<Iterator<org.nd4j.linalg.dataset.DataSet>,org.nd4j.linalg.dataset.DataSet>
| Constructor and Description |
|---|
MiniBatchFunction(int batchSize) |
| Modifier and Type | Method and Description |
|---|---|
Iterable<org.nd4j.linalg.dataset.DataSet> |
call(Iterator<org.nd4j.linalg.dataset.DataSet> dataSetIterator) |
Copyright © 2016. All Rights Reserved.