def draw_model(model):
    from IPython.display import SVG
    from keras.utils.visualize_util import model_to_dot
    from keras.utils.visualize_util import plot

    #graph = to_graph(model, show_shape=True)
    #graph.write_png("UFCNN_1.png")

    SVG(model_to_dot(model).create(prog='dot', format='svg'))
    plot(model, to_file='UFCNN_1.png')