org.deeplearning4j.datasets.rearrange

Class LocalUnstructuredDataFormatter

java.lang.Object

org.deeplearning4j.datasets.rearrange.LocalUnstructuredDataFormatter

public class LocalUnstructuredDataFormatter
extends Object

Rearrange an unstructured dataset in to split test/train on the file system

Author:

Adam Gibson

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	LocalUnstructuredDataFormatter.LabelingType

Constructor Summary

Constructors

Constructor and Description
LocalUnstructuredDataFormatter(File destinationRootDir, File rootDir, LocalUnstructuredDataFormatter.LabelingType labelingType, double percentTrain)

Method Summary

Modifier and Type	Method and Description
String	getNameLabel(String path)
String	getNewDestination(String path, boolean train)
int	getNumExamplesTotal()
int	getNumExamplesToTrainOn()
int	getNumTestExamples()
String	getPathLabel(String path)
File	getTest()
File	getTrain()
void	rearrange()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

LocalUnstructuredDataFormatter

public LocalUnstructuredDataFormatter(File destinationRootDir,
                                      File rootDir,
                                      LocalUnstructuredDataFormatter.LabelingType labelingType,
                                      double percentTrain)

Parameters:

destinationRootDir - the destination root directory

rootDir - the root directory of the original data

labelingType - the labeling type to use (NAME/Label)

percentTrain - the percent train to hold out

Method Detail

rearrange

public void rearrange()

getNewDestination

public String getNewDestination(String path,
                                boolean train)

getPathLabel

public String getPathLabel(String path)

getNameLabel

public String getNameLabel(String path)

getNumExamplesTotal

public int getNumExamplesTotal()

getNumExamplesToTrainOn

public int getNumExamplesToTrainOn()

getNumTestExamples

public int getNumTestExamples()

getTest

public File getTest()

getTrain

public File getTrain()