diff --git a/keras/models/Topology.scala b/keras/models/Topology.scala index bd49a8a41c9..9461410fee9 100644 --- a/keras/models/Topology.scala +++ b/keras/models/Topology.scala @@ -1349,6 +1349,21 @@ private[zoo] class InternalDistriOptimizer[T: ClassTag] ( this } + def getTrainSummary(tag: String): Array[(Long, Float, Double)] = { + if (this.trainSummary isDefined) { + this.trainSummary.get.readScalar(tag) + } else { + null + } + } + + def getValidationSummary(tag: String): Array[(Long, Float, Double)] = { + if (this.validationSummary isDefined) { + this.validationSummary.get.readScalar(tag) + } else { + null + } + } override def train( trainSet: FeatureSet[MiniBatch[T]],