remove keras plotted model

orbxball · Apr 24, 2017 · 97f57f4 · 97f57f4
1 parent 4d69edd
commit 97f57f4
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/hw3/test_cnn.py b/hw3/test_cnn.py
@@ -1,7 +1,6 @@
 #!/usr/bin/env python
 import sys, os
 import numpy as np
-import keras
 from keras.models import load_model
 from keras.utils import plot_model
 
@@ -30,7 +29,7 @@ def ensure_dir(file_path):
 model = load_model(model_name)
 
 # Plot model
-plot_model(model,to_file='cnn_model.png')
+# plot_model(model,to_file='cnn_model.png')
 
 # Predict the test data
 X = X.reshape(X.shape[0], height, width, 1)