|
Java example source code file (ParameterAveragingTrainingResult.java)
The ParameterAveragingTrainingResult.java Java example source codepackage org.deeplearning4j.spark.impl.paramavg; import lombok.Data; import org.deeplearning4j.nn.api.Updater; import org.deeplearning4j.nn.updater.graph.ComputationGraphUpdater; import org.deeplearning4j.spark.api.TrainingResult; import org.deeplearning4j.spark.api.stats.SparkTrainingStats; import org.nd4j.linalg.api.ndarray.INDArray; /** * The results (parameters, optional updaters) returned by a {@link ParameterAveragingTrainingWorker} to the * {@link ParameterAveragingTrainingMaster} * * @author Alex Black */ @Data public class ParameterAveragingTrainingResult implements TrainingResult { private final INDArray parameters; private final Updater updater; private final ComputationGraphUpdater graphUpdater; private final double score; private SparkTrainingStats sparkTrainingStats; public ParameterAveragingTrainingResult(INDArray parameters, Updater updater, double score) { this(parameters, updater, score, null); } public ParameterAveragingTrainingResult(INDArray parameters, ComputationGraphUpdater updater, double score) { this(parameters, null, updater, score, null); } public ParameterAveragingTrainingResult(INDArray parameters, Updater updater, double score, SparkTrainingStats sparkTrainingStats) { this(parameters, updater, null, score, sparkTrainingStats); } public ParameterAveragingTrainingResult(INDArray parameters, ComputationGraphUpdater updater, double score, SparkTrainingStats sparkTrainingStats) { this(parameters, null, updater, score, sparkTrainingStats); } public ParameterAveragingTrainingResult(INDArray parameters, Updater updater, ComputationGraphUpdater graphUpdater, double score, SparkTrainingStats sparkTrainingStats) { this.parameters = parameters; this.updater = updater; this.graphUpdater = graphUpdater; this.score = score; this.sparkTrainingStats = sparkTrainingStats; } @Override public void setStats(SparkTrainingStats sparkTrainingStats) { this.sparkTrainingStats = sparkTrainingStats; } } Other Java examples (source code examples)Here is a short list of links related to this Java ParameterAveragingTrainingResult.java source code file: |
... this post is sponsored by my books ... | |
#1 New Release! |
FP Best Seller |
Copyright 1998-2021 Alvin Alexander, alvinalexander.com
All Rights Reserved.
A percentage of advertising revenue from
pages under the /java/jwarehouse
URI on this website is
paid back to open source projects.