From 100b11128fb0d6766018cf650f74ae086499a746 Mon Sep 17 00:00:00 2001
From: jenniew <jenniewang123@gmail.com>
Date: Tue, 11 Aug 2020 15:49:38 -0700
Subject: [PATCH] Add tensorboard support (#2667)

* add tensorboard support

* fix style

* fix style2

* fix test
---
 .../bigdl/dllib/keras/models/Topology.scala       | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/scala/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/keras/models/Topology.scala b/scala/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/keras/models/Topology.scala
index bd49a8a41c9..9461410fee9 100644
--- a/scala/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/keras/models/Topology.scala
+++ b/scala/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/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]],