public static class OutputLayerTrainingEvaluator.Builder extends Object
Constructor and Description |
---|
Builder() |
Modifier and Type | Method and Description |
---|---|
OutputLayerTrainingEvaluator.Builder |
bestLoss(double bestLoss) |
OutputLayerTrainingEvaluator |
build() |
OutputLayerTrainingEvaluator.Builder |
improvementThreshold(double improvementThreshold) |
OutputLayerTrainingEvaluator.Builder |
miniBatchSize(int miniBatchSize) |
OutputLayerTrainingEvaluator.Builder |
patience(double patience) |
OutputLayerTrainingEvaluator.Builder |
patienceIncrease(double patienceIncrease) |
OutputLayerTrainingEvaluator.Builder |
testSet(org.nd4j.linalg.dataset.DataSet testSet) |
OutputLayerTrainingEvaluator.Builder |
validationEpochs(int validationEpochs) |
OutputLayerTrainingEvaluator.Builder |
withNetwork(MultiLayerNetwork network) |
public OutputLayerTrainingEvaluator.Builder withNetwork(MultiLayerNetwork network)
public OutputLayerTrainingEvaluator.Builder patience(double patience)
public OutputLayerTrainingEvaluator.Builder patienceIncrease(double patienceIncrease)
public OutputLayerTrainingEvaluator.Builder bestLoss(double bestLoss)
public OutputLayerTrainingEvaluator.Builder validationEpochs(int validationEpochs)
public OutputLayerTrainingEvaluator.Builder testSet(org.nd4j.linalg.dataset.DataSet testSet)
public OutputLayerTrainingEvaluator.Builder miniBatchSize(int miniBatchSize)
public OutputLayerTrainingEvaluator.Builder improvementThreshold(double improvementThreshold)
public OutputLayerTrainingEvaluator build()
Copyright © 2015. All rights reserved.