提交 dd827a20 编写于 作者: F fengjiayi

update README.md and add dnn_net.png

上级 868d4cb9
...@@ -6,3 +6,7 @@ ...@@ -6,3 +6,7 @@
## DNN模型结构 ## DNN模型结构
DNN的模型结构入下图所示: DNN的模型结构入下图所示:
<p align="center">
<img src="images/dnn_net.png" width = "80%" align="center"/><br/>
图1. DNN文本分类模型
</p>
...@@ -18,14 +18,18 @@ import gzip ...@@ -18,14 +18,18 @@ import gzip
def fc_net(input_dim, class_dim=2, emb_dim=256): def fc_net(input_dim, class_dim=2, emb_dim=256):
# input layers
data = paddle.layer.data("word", data = paddle.layer.data("word",
paddle.data_type.integer_value_sequence(input_dim)) paddle.data_type.integer_value_sequence(input_dim))
lbl = paddle.layer.data("label", paddle.data_type.integer_value(class_dim)) lbl = paddle.layer.data("label", paddle.data_type.integer_value(class_dim))
# emdedding layer
emb = paddle.layer.embedding(input=data, size=emb_dim) emb = paddle.layer.embedding(input=data, size=emb_dim)
# max pooling
seq_pool = paddle.layer.pooling( seq_pool = paddle.layer.pooling(
input=emb, pooling_type=paddle.pooling.Max()) input=emb, pooling_type=paddle.pooling.Max())
# two hidden layers
hd1 = paddle.layer.fc( hd1 = paddle.layer.fc(
input=seq_pool, input=seq_pool,
size=128, size=128,
...@@ -37,6 +41,7 @@ def fc_net(input_dim, class_dim=2, emb_dim=256): ...@@ -37,6 +41,7 @@ def fc_net(input_dim, class_dim=2, emb_dim=256):
act=paddle.activation.Tanh(), act=paddle.activation.Tanh(),
param_attr=paddle.attr.Param(initial_std=0.01)) param_attr=paddle.attr.Param(initial_std=0.01))
# output layer
output = paddle.layer.fc( output = paddle.layer.fc(
input=hd2, input=hd2,
size=class_dim, size=class_dim,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册