public class ParameterAveragingTrainingResult extends Object implements TrainingResult
ParameterAveragingTrainingWorker
to the
ParameterAveragingTrainingMaster
Constructor and Description |
---|
ParameterAveragingTrainingResult(INDArray parameters,
INDArray updaterState,
double score,
Collection<StorageMetaData> listenerMetaData,
Collection<Persistable> listenerStaticInfo,
Collection<Persistable> listenerUpdates) |
ParameterAveragingTrainingResult(INDArray parameters,
INDArray updaterState,
double score,
SparkTrainingStats sparkTrainingStats,
Collection<StorageMetaData> listenerMetaData,
Collection<Persistable> listenerStaticInfo,
Collection<Persistable> listenerUpdates) |
public ParameterAveragingTrainingResult(INDArray parameters, INDArray updaterState, double score, Collection<StorageMetaData> listenerMetaData, Collection<Persistable> listenerStaticInfo, Collection<Persistable> listenerUpdates)
public ParameterAveragingTrainingResult(INDArray parameters, INDArray updaterState, double score, SparkTrainingStats sparkTrainingStats, Collection<StorageMetaData> listenerMetaData, Collection<Persistable> listenerStaticInfo, Collection<Persistable> listenerUpdates)
public void setStats(SparkTrainingStats sparkTrainingStats)
setStats
in interface TrainingResult
Copyright © 2020. All rights reserved.