【发布时间】: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 版。
【问题讨论】: