|
Java example source code file (AbstractDataSetIteratorTest.java)
The AbstractDataSetIteratorTest.java Java example source codepackage org.deeplearning4j.datasets.iterator; import org.apache.commons.lang3.RandomUtils; import org.deeplearning4j.berkeley.Pair; import org.junit.Test; import org.nd4j.linalg.api.ndarray.INDArray; import org.nd4j.linalg.dataset.DataSet; import java.util.Iterator; import java.util.concurrent.atomic.AtomicInteger; import static org.junit.Assert.*; /** * Created by raver on 16.06.2016. */ public class AbstractDataSetIteratorTest { @Test public void next() throws Exception { int numFeatures = 128; int batchSize = 10; int numRows = 1000; AtomicInteger cnt = new AtomicInteger(0); FloatsDataSetIterator iterator = new FloatsDataSetIterator(floatIterable(numRows, numFeatures), batchSize); assertTrue(iterator.hasNext()); while (iterator.hasNext()) { DataSet dataSet = iterator.next(); INDArray features = dataSet.getFeatures(); assertEquals(batchSize, features.rows()); assertEquals(numFeatures, features.columns()); cnt.incrementAndGet(); } assertEquals(numRows/batchSize, cnt.get()); } protected static Iterable<Pair Other Java examples (source code examples)Here is a short list of links related to this Java AbstractDataSetIteratorTest.java source code file: |
... this post is sponsored by my books ... | |
#1 New Release! |
FP Best Seller |
Copyright 1998-2021 Alvin Alexander, alvinalexander.com
All Rights Reserved.
A percentage of advertising revenue from
pages under the /java/jwarehouse
URI on this website is
paid back to open source projects.