public class HyperParameterSearch extends Object
Modifier and Type | Field and Description |
---|---|
static String |
DIVIDE_AND_CONQUER |
static String |
GRID |
static String |
RANDOM |
Constructor and Description |
---|
HyperParameterSearch() |
public static final String GRID
public static final String RANDOM
public static final String DIVIDE_AND_CONQUER
public HyperParameterSearch trainingSet(javax.visrec.ml.data.DataSet trainingSet)
public HyperParameterSearch testSet(javax.visrec.ml.data.DataSet testSet)
public HyperParameterSearch evaluator(javax.visrec.ml.eval.Evaluator<NeuralNetwork,javax.visrec.ml.data.DataSet<? extends MLDataItem>> evaluator)
public HyperParameterSearch trainingListener(TrainingListener trainingListener)
public <T> HyperParameterSearch param(String name, T value)
public <T> HyperParameterSearch param(String name, List<T> values)
public <T> HyperParameterSearch param(String name, Range<T> range, String searchStrategy)
public Parameters getParameters()
public final NeuralNetwork getNetwork()
public final javax.visrec.ml.data.DataSet<?> getTrainingSet()
public final javax.visrec.ml.eval.Evaluator<NeuralNetwork,javax.visrec.ml.data.DataSet<? extends MLDataItem>> getEvaluator()
public void run()
public List<Parameters.ParameterCombination> getSearchSpace()
public List<TrainingResult> getResults()
public HyperParameterSearch randomSeed(long randomSeed)
public HyperParameterSearch networkFactory(NetworkFactory<?> networkFactory)
Copyright © 2022. All rights reserved.