plot_network

plot_network(input_shape, network, file_name=None, show_shapes=True, show_layer_names=False, show_layer_activations=True, dpi=96)[source]

绘制网络及其输入输出形状和激活。

无论输出任务如何,网络都会被绘制出来。 需要注意的是,为了能够使用此功能, 环境中应安装pydot包,并且系统中应安装graphviz。 当网络是基于自动编码器的模型时,该功能也有效, 并且该功能会将输出文件分成2个文件, 第一个用于编码器,第二个用于解码器。

Parameters:
input_shapetuple

输入层接收的数据的形状。

networkan element from the network module

示例:FCNNetwork().

file_namestr default = None (“model.pdf”)

绘图图像的文件名,不包括扩展名。

show_shapesbool, default = True

是否显示形状信息。

show_layer_namesbool, default = False

是否显示图层名称。

show_layer_activationsbool, default = True

显示层激活 (仅适用于具有激活属性的层)。

dpiint, default = 96

每英寸点数。

Returns:
None