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, setListenerpublic 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.