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<org.nd4j.linalg.dataset.DataSet> train) |
SparkEarlyStoppingTrainer(org.apache.spark.api.java.JavaSparkContext sc,
TrainingMaster trainingMaster,
EarlyStoppingConfiguration<MultiLayerNetwork> esConfig,
MultiLayerNetwork net,
org.apache.spark.api.java.JavaRDD<org.nd4j.linalg.dataset.DataSet> train,
EarlyStoppingListener<MultiLayerNetwork> listener) |
SparkEarlyStoppingTrainer(org.apache.spark.SparkContext sc,
TrainingMaster trainingMaster,
EarlyStoppingConfiguration<MultiLayerNetwork> esConfig,
MultiLayerNetwork net,
org.apache.spark.api.java.JavaRDD<org.nd4j.linalg.dataset.DataSet> train) |
SparkEarlyStoppingTrainer(org.apache.spark.SparkContext sc,
TrainingMaster trainingMaster,
EarlyStoppingConfiguration<MultiLayerNetwork> esConfig,
MultiLayerNetwork net,
org.apache.spark.api.java.JavaRDD<org.nd4j.linalg.dataset.DataSet> train,
EarlyStoppingListener<MultiLayerNetwork> listener) |
Modifier and Type | Method and Description |
---|---|
protected void |
fit(org.apache.spark.api.java.JavaRDD<org.nd4j.linalg.dataset.DataSet> data) |
protected void |
fitMulti(org.apache.spark.api.java.JavaRDD<org.nd4j.linalg.dataset.api.MultiDataSet> data) |
protected double |
getScore() |
fit, setListener
public SparkEarlyStoppingTrainer(org.apache.spark.SparkContext sc, TrainingMaster trainingMaster, EarlyStoppingConfiguration<MultiLayerNetwork> esConfig, MultiLayerNetwork net, org.apache.spark.api.java.JavaRDD<org.nd4j.linalg.dataset.DataSet> train)
public SparkEarlyStoppingTrainer(org.apache.spark.api.java.JavaSparkContext sc, TrainingMaster trainingMaster, EarlyStoppingConfiguration<MultiLayerNetwork> esConfig, MultiLayerNetwork net, org.apache.spark.api.java.JavaRDD<org.nd4j.linalg.dataset.DataSet> train)
public SparkEarlyStoppingTrainer(org.apache.spark.SparkContext sc, TrainingMaster trainingMaster, EarlyStoppingConfiguration<MultiLayerNetwork> esConfig, MultiLayerNetwork net, org.apache.spark.api.java.JavaRDD<org.nd4j.linalg.dataset.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<org.nd4j.linalg.dataset.DataSet> train, EarlyStoppingListener<MultiLayerNetwork> listener)
protected void fit(org.apache.spark.api.java.JavaRDD<org.nd4j.linalg.dataset.DataSet> data)
fit
in class BaseSparkEarlyStoppingTrainer<MultiLayerNetwork>
protected void fitMulti(org.apache.spark.api.java.JavaRDD<org.nd4j.linalg.dataset.api.MultiDataSet> data)
fitMulti
in class BaseSparkEarlyStoppingTrainer<MultiLayerNetwork>
protected double getScore()
getScore
in class BaseSparkEarlyStoppingTrainer<MultiLayerNetwork>