Modifier and Type | Field and Description |
---|---|
protected EarlyStoppingConfiguration<ComputationGraph> |
ComputationGraphSpace.earlyStoppingConfiguration |
protected EarlyStoppingConfiguration<ComputationGraph> |
ComputationGraphSpace.Builder.earlyStoppingConfiguration |
protected EarlyStoppingConfiguration<MultiLayerNetwork> |
MultiLayerSpace.earlyStoppingConfiguration |
protected EarlyStoppingConfiguration<MultiLayerNetwork> |
MultiLayerSpace.Builder.earlyStoppingConfiguration |
Modifier and Type | Method and Description |
---|---|
ComputationGraphSpace.Builder |
ComputationGraphSpace.Builder.earlyStoppingConfiguration(EarlyStoppingConfiguration<ComputationGraph> earlyStoppingConfiguration)
Early stopping configuration (optional).
|
MultiLayerSpace.Builder |
MultiLayerSpace.Builder.earlyStoppingConfiguration(EarlyStoppingConfiguration<MultiLayerNetwork> earlyStoppingConfiguration)
Early stopping configuration (optional).
|
Modifier and Type | Method and Description |
---|---|
EarlyStoppingConfiguration<T> |
EarlyStoppingConfiguration.Builder.build()
Create the early stopping configuration
|
Modifier and Type | Method and Description |
---|---|
void |
EarlyStoppingListener.onEpoch(int epochNum,
double score,
EarlyStoppingConfiguration<T> esConfig,
T net)
Method that is called at the end of each epoch completed during early stopping training
|
void |
EarlyStoppingListener.onStart(EarlyStoppingConfiguration<T> esConfig,
T net)
Method to be called when early stopping training is first started
|
Modifier and Type | Field and Description |
---|---|
protected EarlyStoppingConfiguration<T> |
BaseEarlyStoppingTrainer.esConfig |
Modifier and Type | Field and Description |
---|---|
protected EarlyStoppingConfiguration<T> |
EarlyStoppingParallelTrainer.esConfig |
Constructor and Description |
---|
EarlyStoppingParallelTrainer(EarlyStoppingConfiguration<T> earlyStoppingConfiguration,
T model,
DataSetIterator train,
MultiDataSetIterator trainMulti,
EarlyStoppingListener<T> listener,
int workers,
int prefetchBuffer,
int averagingFrequency) |
EarlyStoppingParallelTrainer(EarlyStoppingConfiguration<T> earlyStoppingConfiguration,
T model,
DataSetIterator train,
MultiDataSetIterator trainMulti,
EarlyStoppingListener<T> listener,
int workers,
int prefetchBuffer,
int averagingFrequency,
boolean reportScoreAfterAveraging,
boolean useLegacyAveraging) |
EarlyStoppingParallelTrainer(EarlyStoppingConfiguration<T> earlyStoppingConfiguration,
T model,
DataSetIterator train,
MultiDataSetIterator trainMulti,
int workers,
int prefetchBuffer,
int averagingFrequency) |
Constructor and Description |
---|
BaseSparkEarlyStoppingTrainer(org.apache.spark.api.java.JavaSparkContext sc,
EarlyStoppingConfiguration<T> esConfig,
T net,
org.apache.spark.api.java.JavaRDD<DataSet> train,
org.apache.spark.api.java.JavaRDD<MultiDataSet> trainMulti,
EarlyStoppingListener<T> listener) |
SparkEarlyStoppingGraphTrainer(org.apache.spark.api.java.JavaSparkContext sc,
TrainingMaster trainingMaster,
EarlyStoppingConfiguration<ComputationGraph> esConfig,
ComputationGraph net,
org.apache.spark.api.java.JavaRDD<MultiDataSet> train) |
SparkEarlyStoppingGraphTrainer(org.apache.spark.api.java.JavaSparkContext sc,
TrainingMaster trainingMaster,
EarlyStoppingConfiguration<ComputationGraph> esConfig,
ComputationGraph net,
org.apache.spark.api.java.JavaRDD<MultiDataSet> train,
EarlyStoppingListener<ComputationGraph> listener) |
SparkEarlyStoppingGraphTrainer(org.apache.spark.api.java.JavaSparkContext sc,
TrainingMaster trainingMaster,
EarlyStoppingConfiguration<ComputationGraph> esConfig,
ComputationGraph net,
org.apache.spark.api.java.JavaRDD<MultiDataSet> train,
int examplesPerFit,
int totalExamples) |
SparkEarlyStoppingGraphTrainer(org.apache.spark.SparkContext sc,
TrainingMaster trainingMaster,
EarlyStoppingConfiguration<ComputationGraph> esConfig,
ComputationGraph net,
org.apache.spark.api.java.JavaRDD<MultiDataSet> train) |
SparkEarlyStoppingGraphTrainer(org.apache.spark.SparkContext sc,
TrainingMaster trainingMaster,
EarlyStoppingConfiguration<ComputationGraph> esConfig,
ComputationGraph net,
org.apache.spark.api.java.JavaRDD<MultiDataSet> train,
int examplesPerFit,
int totalExamples) |
SparkEarlyStoppingTrainer(org.apache.spark.api.java.JavaSparkContext sc,
TrainingMaster trainingMaster,
EarlyStoppingConfiguration<MultiLayerNetwork> esConfig,
MultiLayerNetwork net,
org.apache.spark.api.java.JavaRDD<DataSet> train) |
SparkEarlyStoppingTrainer(org.apache.spark.api.java.JavaSparkContext sc,
TrainingMaster trainingMaster,
EarlyStoppingConfiguration<MultiLayerNetwork> esConfig,
MultiLayerNetwork net,
org.apache.spark.api.java.JavaRDD<DataSet> train,
EarlyStoppingListener<MultiLayerNetwork> listener) |
SparkEarlyStoppingTrainer(org.apache.spark.SparkContext sc,
TrainingMaster trainingMaster,
EarlyStoppingConfiguration<MultiLayerNetwork> esConfig,
MultiLayerNetwork net,
org.apache.spark.api.java.JavaRDD<DataSet> train) |
SparkEarlyStoppingTrainer(org.apache.spark.SparkContext sc,
TrainingMaster trainingMaster,
EarlyStoppingConfiguration<MultiLayerNetwork> esConfig,
MultiLayerNetwork net,
org.apache.spark.api.java.JavaRDD<DataSet> train,
EarlyStoppingListener<MultiLayerNetwork> listener) |
Copyright © 2020. All rights reserved.