Java example source code file (LocalUnstructuredDataFormatterTest.java)

This example Java source code file (LocalUnstructuredDataFormatterTest.java) is included in the alvinalexander.com "Java Source Code Warehouse" project. The intent of this project is to help you "Learn Java by Example" ^TM.

Learn more about this Java project at its project page.

Java - Java tags/keywords

exception, file, iterator, lfwdatasetiterator, localunstructureddataformatter, localunstructureddataformattertest, util

The LocalUnstructuredDataFormatterTest.java Java example source code

package org.deeplearning4j.datasets.rearrange;

import org.apache.commons.io.FileUtils;
import org.deeplearning4j.datasets.iterator.impl.LFWDataSetIterator;
import org.junit.Test;

import java.io.File;
import java.util.Iterator;

import static org.junit.Assert.*;


/**
 * Created by agibsonccc on 9/11/15.
 */
public class LocalUnstructuredDataFormatterTest {

    @Test
    public void testRearrange() throws Exception {
        //ensure exists
        new LFWDataSetIterator(new int[] {10,28,28}).next();
        File destinationDir = new File(System.getProperty("user.home"),"rearrangedlfw");
        if(destinationDir.exists())
            FileUtils.deleteDirectory(destinationDir);
        LocalUnstructuredDataFormatter formatter = new LocalUnstructuredDataFormatter(destinationDir,new File(System.getProperty("user.home"),"lfw"), LocalUnstructuredDataFormatter.LabelingType.DIRECTORY,0.8);
        formatter.rearrange();
        //train and test in the split directory
        File splitRoot = new File(destinationDir,"split");
        assertEquals(2,splitRoot.listFiles().length);
        Iterator<File> files = FileUtils.iterateFiles(splitRoot,null,true);
        int count = 0;
        while(files.hasNext()) {
            files.next();
            count++;
        }

        Iterator<File> trainFiles = FileUtils.iterateFiles(new File(splitRoot,"train"),null,true);
        int trainFilesCount = 0;
        while(trainFiles.hasNext()) {
            trainFiles.next();
            trainFilesCount++;
        }

        //assert the number of files is the same in the split test train as the original dataset
        assertEquals(formatter.getNumExamplesTotal(),count);
        assertEquals(formatter.getNumExamplesToTrainOn(),trainFilesCount);

        Iterator<File> testFiles = FileUtils.iterateFiles(new File(splitRoot,"test"),null,true);
        int testFilesCount = 0;
        while(testFiles.hasNext()) {
            testFiles.next();
            testFilesCount++;
        }
        assertEquals(formatter.getNumTestExamples(),testFilesCount);

    }

}