【问题标题】:Theano graph printing in KerasKeras 中的 Theano 图形打印
【发布时间】:2016-08-20 07:25:53
【问题描述】:

有没有办法在 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 profiler 的输出。

Keras 1.0 版,Theano 0.8.1 版。

【问题讨论】:

    标签: python theano keras


    【解决方案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]
    

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2017-06-27
      • 2021-07-03
      • 2013-02-08
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多