org.deeplearning4j.spark.impl.paramavg

Class ParameterAveragingTrainingResult

java.lang.Object

org.deeplearning4j.spark.impl.paramavg.ParameterAveragingTrainingResult

All Implemented Interfaces:

TrainingResult

public class ParameterAveragingTrainingResult
extends java.lang.Object
implements TrainingResult

The results (parameters, optional updaters) returned by a ParameterAveragingTrainingWorker to the ParameterAveragingTrainingMaster

Constructor Summary

Constructors

Constructor and Description
ParameterAveragingTrainingResult(org.nd4j.linalg.api.ndarray.INDArray parameters, org.nd4j.linalg.api.ndarray.INDArray updaterState, double score, java.util.Collection<StorageMetaData> listenerMetaData, java.util.Collection<Persistable> listenerStaticInfo, java.util.Collection<Persistable> listenerUpdates)
ParameterAveragingTrainingResult(org.nd4j.linalg.api.ndarray.INDArray parameters, org.nd4j.linalg.api.ndarray.INDArray updaterState, double score, SparkTrainingStats sparkTrainingStats, java.util.Collection<StorageMetaData> listenerMetaData, java.util.Collection<Persistable> listenerStaticInfo, java.util.Collection<Persistable> listenerUpdates)

Method Summary

Modifier and Type	Method and Description
void	setStats(SparkTrainingStats sparkTrainingStats)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

ParameterAveragingTrainingResult

public ParameterAveragingTrainingResult(org.nd4j.linalg.api.ndarray.INDArray parameters,
                                        org.nd4j.linalg.api.ndarray.INDArray updaterState,
                                        double score,
                                        java.util.Collection<StorageMetaData> listenerMetaData,
                                        java.util.Collection<Persistable> listenerStaticInfo,
                                        java.util.Collection<Persistable> listenerUpdates)

ParameterAveragingTrainingResult

public ParameterAveragingTrainingResult(org.nd4j.linalg.api.ndarray.INDArray parameters,
                                        org.nd4j.linalg.api.ndarray.INDArray updaterState,
                                        double score,
                                        SparkTrainingStats sparkTrainingStats,
                                        java.util.Collection<StorageMetaData> listenerMetaData,
                                        java.util.Collection<Persistable> listenerStaticInfo,
                                        java.util.Collection<Persistable> listenerUpdates)

Method Detail

setStats

public void setStats(SparkTrainingStats sparkTrainingStats)

Specified by:

setStats in interface TrainingResult