[Scala] hide _move_var _move_mean _beta _gamma for visualization (#6555)

* [Scala] hide _move_var _move_mean _beta _gamma for visualization * fix code style
apache · Jun 3, 2017 · 673a6db · 673a6db
1 parent 13c1463
commit 673a6db
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/scala-package/core/src/main/scala/ml/dmlc/mxnet/Visualization.scala b/scala-package/core/src/main/scala/ml/dmlc/mxnet/Visualization.scala
@@ -216,8 +216,9 @@ object Visualization {
 
     // Internal helper to figure out if node should be hidden with hide_weights
     def looksLikeWeight(name: String): Boolean = {
-      if (name.endsWith("_weight") || name.endsWith("_bias")) true
-      else false
+      if (name.endsWith("_weight") || name.endsWith("_bias")
+          || name.endsWith("_beta") || name.endsWith("_gamma")
+          || name.endsWith("_moving_var") || name.endsWith("_moving_mean")) { true } else { false }
     }
 
     // make nodes