public class SparkEarlyStoppingTrainer extends BaseSparkEarlyStoppingTrainer<MultiLayerNetwork>
MultiLayerNetwork
Constructor and Description |
---|
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) |
Modifier and Type | Method and Description |
---|---|
protected void |
fit(org.apache.spark.api.java.JavaRDD<DataSet> data) |
protected void |
fitMulti(org.apache.spark.api.java.JavaRDD<MultiDataSet> data) |
protected double |
getScore() |
EarlyStoppingResult<MultiLayerNetwork> |
pretrain() |
fit, setListener
public SparkEarlyStoppingTrainer(org.apache.spark.SparkContext sc, TrainingMaster trainingMaster, EarlyStoppingConfiguration<MultiLayerNetwork> esConfig, MultiLayerNetwork net, org.apache.spark.api.java.JavaRDD<DataSet> train)
public SparkEarlyStoppingTrainer(org.apache.spark.api.java.JavaSparkContext sc, TrainingMaster trainingMaster, EarlyStoppingConfiguration<MultiLayerNetwork> esConfig, MultiLayerNetwork net, org.apache.spark.api.java.JavaRDD<DataSet> train)
public SparkEarlyStoppingTrainer(org.apache.spark.SparkContext sc, TrainingMaster trainingMaster, EarlyStoppingConfiguration<MultiLayerNetwork> esConfig, MultiLayerNetwork net, org.apache.spark.api.java.JavaRDD<DataSet> train, EarlyStoppingListener<MultiLayerNetwork> listener)
public 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)
protected void fit(org.apache.spark.api.java.JavaRDD<DataSet> data)
fit
in class BaseSparkEarlyStoppingTrainer<MultiLayerNetwork>
protected void fitMulti(org.apache.spark.api.java.JavaRDD<MultiDataSet> data)
fitMulti
in class BaseSparkEarlyStoppingTrainer<MultiLayerNetwork>
protected double getScore()
getScore
in class BaseSparkEarlyStoppingTrainer<MultiLayerNetwork>
public EarlyStoppingResult<MultiLayerNetwork> pretrain()
Copyright © 2020. All rights reserved.