|
Java example source code file (MiniBatchTests.java)
The MiniBatchTests.java Java example source code/* * * * Copyright 2015 Skymind,Inc. * * * * Licensed under the Apache License, Version 2.0 (the "License"); * * you may not use this file except in compliance with the License. * * You may obtain a copy of the License at * * * * http://www.apache.org/licenses/LICENSE-2.0 * * * * Unless required by applicable law or agreed to in writing, software * * distributed under the License is distributed on an "AS IS" BASIS, * * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * * See the License for the specific language governing permissions and * * limitations under the License. * */ package org.deeplearning4j.spark.canova; import org.apache.spark.api.java.JavaRDD; import org.apache.spark.api.java.function.Function; import org.canova.api.records.reader.impl.SVMLightRecordReader; import org.deeplearning4j.spark.BaseSparkTest; import org.junit.Test; import org.nd4j.linalg.dataset.DataSet; import org.nd4j.linalg.io.ClassPathResource; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import static org.junit.Assert.*; /** * Handle dividing things up by mini batch */ public class MiniBatchTests extends BaseSparkTest { private static final Logger log = LoggerFactory.getLogger(MiniBatchTests.class); @Test public void testMiniBatches() throws Exception { log.info("Setting up Spark Context..."); JavaRDD<String> lines = sc.textFile(new ClassPathResource("svmLight/iris_svmLight_0.txt").getFile().toURI().toString()).cache(); long count = lines.count(); assertEquals(300,count); // gotta map this to a Matrix/INDArray JavaRDD<DataSet> points = lines.map(new RecordReaderFunction(new SVMLightRecordReader(), 4, 3)).cache(); count = points.count(); assertEquals(300,count); JavaRDD<DataSet> miniBatches = new RDDMiniBatches(10,points).miniBatchesJava(); count = miniBatches.count(); assertEquals(30,count); miniBatches.map(new DataSetAssertionFunction()); } public static class DataSetAssertionFunction implements Function<DataSet,Object> { @Override public Object call(DataSet dataSet) throws Exception { assertTrue(dataSet.getFeatureMatrix().columns() == 150); assertTrue(dataSet.numExamples() == 30); return null; } } } Other Java examples (source code examples)Here is a short list of links related to this Java MiniBatchTests.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.