From fd2d2a754e1168538b1169cdc0ea18312e05e9f1 Mon Sep 17 00:00:00 2001
From: Jui Shah <34505328+shahjui2000@users.noreply.github.com>
Date: Fri, 21 Feb 2020 21:02:31 +0530
Subject: [PATCH] [TMVA][PyKeras] Keras Visualization used is outdate (#5014)
* Update GenerateModel.py
keras.utils.visualize_util is outdated
* Update GenerateModel.py
---
tutorials/tmva/keras/GenerateModel.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tutorials/tmva/keras/GenerateModel.py b/tutorials/tmva/keras/GenerateModel.py
index 1c3a292696f..219d3ebbe0a 100755
--- a/tutorials/tmva/keras/GenerateModel.py
+++ b/tutorials/tmva/keras/GenerateModel.py
@@ -14,6 +14,7 @@ from keras.models import Sequential
from keras.layers.core import Dense, Activation
from keras.regularizers import l2
from keras.optimizers import SGD
+from keras.utils import plot_model
# Setup the model here
num_input_nodes = 4
@@ -57,7 +58,6 @@ model.summary()
# Visualize model as graph
try:
- from keras.utils.visualize_util import plot
- plot(model, to_file='model.png', show_shapes=True)
+ plot_model(model, to_file='model.png', show_shapes=True)
except:
print('[INFO] Failed to make model plot')
--
GitLab