Add tensorboard support (intel-analytics#2667)

* add tensorboard support * fix style * fix style2 * fix test
dding3 · Aug 11, 2020 · 7e1ede7 · 7e1ede7
1 parent b897a0b
commit 7e1ede7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pyspark/bigdl/orca/tfpark/tf_optimizer.py b/pyspark/bigdl/orca/tfpark/tf_optimizer.py
@@ -44,6 +44,8 @@ def __init__(self):
 
 class TFValidationMethod(JavaValue):
     def __init__(self, val_method, name, output_indices, label_indices):
+        self.name = name
+        self.val_method = val_method
         JavaValue.__init__(self, None, "float",
                            val_method, name, output_indices, label_indices)
 
@@ -569,7 +571,7 @@ def from_loss(cls, loss, optim_method, session=None, inputs=None, dataset=None,
                 metrics = {}
 
             for i, method in enumerate(val_methods):
-                metrics['bigdl_metirc_' + str(i)] = BigDLMetric(method, val_outputs, val_labels)
+                metrics['bigdl_metric_' + str(i)] = BigDLMetric(method, val_outputs, val_labels)
 
         return TFOptimizer._from_grads(loss, sess, inputs, labels, grads, variables, dataset,
                                        optim_method, clip_norm, clip_value,
@@ -688,7 +690,7 @@ def from_keras(cls, keras_model, dataset,
             val_methods = to_list(bigdl_val_methods)
             bigdl_metrics = {}
             for i, method in enumerate(val_methods):
-                bigdl_metrics['bigdl_metirc_' + str(i)] = BigDLMetric(method,
+                bigdl_metrics['bigdl_metric_' + str(i)] = BigDLMetric(method,
                                                                       val_outputs,
                                                                       val_labels)
             if metrics is None: