Question

我正在使用pybrain构建神经网络。有时情况的图形表示非常有用。

是否可以绘制使用pybrain生成的神经网络的结构？

Answer 1

我认为最初的提问者可能正在寻找类似的东西（就像我一样，虽然没有动画）：http://www.codeproject.com/KB/dotnet/predictor/learn.gif

我认为在这篇文章中或多或少地回答了这个问题： How to visualize a neural network

＆＃34;或多或少＆＃34;因为在每个圆圈内看到单元格引用（A0，A1，A2，B0等等）会很高兴。

但我是Python和神经网络的初学者，如果我错了，请随时纠正我。

盖

Answer 2

如前所述，答案How to visualize a neural network显示了如何使用pyplot绘制简单网络。

以下是针对PyBrain调整此解决方案的方法：

class PybrainNNVisualizer():
    def __init__(self, neural_network):
        """
        :type neural_network: Network
        """
        self.neural_network = neural_network

    def draw(self):
        widest_layer = max([layer.dim for layer in self.neural_network.modules])
        network = NeuralNetwork(widest_layer)
        for layer in self.neural_network.modulesSorted:
            if type(layer) is BiasUnit:
                continue
            network.add_layer(layer.dim)
        network.draw()

用法：

fnn = buildNetwork(4, 8, 1)

PybrainNNVisualizer(fnn).draw()

完整的源代码：https://github.com/AlexP11223/SimplePyBrainNeuralNeutwork/blob/master/nnvisualizer.py

如何绘制byprain神经网络结构

2 个答案: