code: def draw_to_notebook(layers, **kwargs):
      
      from IPython.display import Image
      layers = (layers.get_all_layers() if hasattr(layers, 'get_all_layers')
                else layers)
      dot = make_pydot_graph(layers, **kwargs)
      return Image(dot.create_png())
docs: Draws a network diagram in an IPython notebook
      :parameters:
          - layers : list or NeuralNet instance
              List of layers or the neural net to draw.
          - **kwargs : see the docstring of make_pydot_graph for other options