Add tensorboard support (intel-analytics#2667)

* add tensorboard support * fix style * fix style2 * fix test
Le-Zheng · Aug 11, 2020 · d7a7e37 · d7a7e37
1 parent 0bf5569
commit d7a7e37
Showing 1 changed file with 15 additions and 0 deletions.
diff --git 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]],