test_cost_layers.py 1.1 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13
from paddle.trainer_config_helpers import *

settings(
    learning_rate=1e-4,
    batch_size=1000
)

seq_in = data_layer(name='input', size=200)
labels = data_layer(name='labels', size=5000)

probs = data_layer(name='probs', size=10)
xe_label = data_layer(name='xe-label', size=10)

X
xuwei06 已提交
14
hidden = fc_layer(input=seq_in, size=4)
15
outputs(ctc_layer(input=seq_in, label=labels),
X
xuwei06 已提交
16
        crf_layer(input=hidden,
17 18 19 20 21 22 23 24 25 26
                  label=data_layer(name='crf_label', size=4)),
        rank_cost(left=data_layer(name='left', size=1),
                  right=data_layer(name='right', size=1),
                  label=data_layer(name='label', size=1)),
        lambda_cost(input=data_layer(name='list_feature', size=100),
                    score=data_layer(name='list_scores', size=1)),
        cross_entropy(input=probs, label=xe_label),
        cross_entropy_with_selfnorm(input=probs, label=xe_label),
        huber_cost(input=data_layer(name='huber_probs', size=1),
                   label=data_layer(name='huber_label', size=1)),
X
xuwei06 已提交
27
        multi_binary_label_cross_entropy(input=probs, label=xe_label),
L
Luo Tao 已提交
28
        sum_cost(input=hidden))