shared_fc.py 869 字节
Newer Older
Y
Yu Yang 已提交
1 2
from paddle.trainer_config_helpers import *

Q
qijun 已提交
3
settings(learning_rate=1e-4, batch_size=1000)
Y
Yu Yang 已提交
4 5 6 7 8 9 10

a = data_layer(name='feature_a', size=200)
b = data_layer(name='feature_b', size=200)

fc_param = ParamAttr(name='fc_param', initial_max=1.0, initial_min=-1.0)
bias_param = ParamAttr(name='bias_param', initial_mean=0.0, initial_std=0.0)

Q
qijun 已提交
11 12
softmax_param = ParamAttr(
    name='softmax_param', initial_max=1.0, initial_min=-1.0)
Y
Yu Yang 已提交
13

Q
qijun 已提交
14 15 16 17
hidden_a = fc_layer(
    input=a, size=200, param_attr=fc_param, bias_attr=bias_param)
hidden_b = fc_layer(
    input=b, size=200, param_attr=fc_param, bias_attr=bias_param)
Y
Yu Yang 已提交
18

Q
qijun 已提交
19 20 21 22 23 24
predict = fc_layer(
    input=[hidden_a, hidden_b],
    param_attr=[softmax_param, softmax_param],
    bias_attr=False,
    size=10,
    act=SoftmaxActivation())
Y
Yu Yang 已提交
25

Q
qijun 已提交
26 27 28 29
outputs(
    classification_cost(
        input=predict, label=data_layer(
            name='label', size=10)))