python - Keras 中的 Theano 图形打印

Question

有没有办法在 Keras 中打印出顺序模型的 Theano 图表示？

给定一个像这样的简单模型:

model = Sequential()

model.add(Dense(input_dim=100,
                output_dim=50,
                W_regularizer=l2(0.0001),
                init='he_normal',
                activation='tanh'))

model.add(Dense(input_dim=50,
                output_dim=1,
                W_regularizer=l2(0.0001),
                init='he_normal',
                activation='sigmoid'))

sgd = SGD(lr=0.01, decay=1e-7, momentum=.9)
model.compile(loss='binary_crossentropy', optimizer=sgd, metrics=["accuracy"])

我想运行:

theano.printing.debugprint(model)

给我这样的东西(取自 Theano 教程):

Elemwise{mul,no_inplace} [id A] ''
 |TensorConstant{2.0} [id B]
 |x [id C]

我希望获得图表来帮助理解 Theano 分析器的输出。

Keras 版本 1.0，Theano 版本 0.8.1。

Answer 1

如果其他人想做类似的事情，这是我发现的一种方法。

import theano
for i in model.layers:
    theano.printing.debugprint(i.input)
    theano.printing.debugprint(i.output)

它将给出:

dense_input_2 [id A]
Elemwise{tanh,no_inplace} [id A] ''   
 |Elemwise{add,no_inplace} [id B] ''   
   |dot [id C] ''   
   | |dense_input_2 [id D]
   | |dense_4_W [id E]
   |DimShuffle{x,0} [id F] ''   
     |dense_4_b [id G]
Elemwise{tanh,no_inplace} [id A] ''   
 |Elemwise{add,no_inplace} [id B] ''   
   |dot [id C] ''   
   | |dense_input_2 [id D]
   | |dense_4_W [id E]
   |DimShuffle{x,0} [id F] ''   
     |dense_4_b [id G]
sigmoid [id A] ''   
 |Elemwise{add,no_inplace} [id B] ''   
   |dot [id C] ''   
   | |Elemwise{tanh,no_inplace} [id D] ''   
   | | |Elemwise{add,no_inplace} [id E] ''   
   | |   |dot [id F] ''   
   | |   | |dense_input_2 [id G]
   | |   | |dense_4_W [id H]
   | |   |DimShuffle{x,0} [id I] ''   
   | |     |dense_4_b [id J]
   | |dense_5_W [id K]
   |DimShuffle{x,0} [id L] ''   
     |dense_5_b [id M]