org.deeplearning4j.eval

Class DataSetTester

java.lang.Object

org.deeplearning4j.base.DeepLearningTest

org.deeplearning4j.eval.DataSetTester

public class DataSetTester
extends DeepLearningTest

DataSet runner main class. Basic idea is to feed it an algorithm, dataset, and the number of examples to use. It will then print out f1 scores for each dataset. Note that I need to add WAY more for tuning this yet as far as command line options go.

Author:

Adam Gibson

Constructor Summary

Constructors

Constructor and Description
DataSetTester(String dataset, String algorithm)
DataSetTester(String dataset, String algorithm, Integer numExamples)

Method Summary

Methods

Modifier and Type	Method and Description
static void	main(String[] args)
void	run()

Methods inherited from class org.deeplearning4j.base.DeepLearningTest

getFaces, getFaces, getFacesMatrix, getFirstFaces, getIris, getIris, getMnistExample, getMnistExampleBatch, getMnistExampleBatches

Methods inherited from class java.lang.Object

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

Constructor Detail

DataSetTester

public DataSetTester(String dataset,
             String algorithm,
             Integer numExamples)

DataSetTester

public DataSetTester(String dataset,
             String algorithm)

Method Detail

main

public static void main(String[] args)
                 throws Exception

Parameters:

args -

Throws:

Exception

run

public void run()
         throws Exception

Throws:

Exception