可视化神经网络的网络结构并保存

Posted 2021-05-18 Data+Science+Insight

可视化神经网络的网络结构并保存

#导入基础包和方法

# Load libraries
from keras import models
from keras import layers
from IPython.display import SVG
from keras.utils.vis_utils import model_to_dot
from keras.utils import plot_model

#构建网络结构

# Start neural network
network = models.Sequential()

# Add fully connected layer with a ReLU activation function
network.add(layers.Dense(units=16, activation=\'relu\', input_shape=(10,)))

# Add fully connected layer with a ReLU activation function
network.add(layers.Dense(units=16, activation=\'relu\'))

# Add fully connected layer with a sigmoid activation function
network.add(layers.Dense(units=1, activation=\'sigmoid\'))

#可视化网络结构

# Visualize network architecture
SVG(model_to_dot(network, show_shapes=True