test_cost_layers.py 1.3 KB
Newer Older
1 2
from paddle.trainer_config_helpers import *

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

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 已提交
11
hidden = fc_layer(input=seq_in, size=4)
Q
qijun 已提交
12 13 14
outputs(
    ctc_layer(
        input=seq_in, label=labels),
15 16
    warp_ctc_layer(
        input=seq_in, label=labels, blank=0),
Q
qijun 已提交
17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42
    crf_layer(
        input=hidden, 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)),
    multi_binary_label_cross_entropy(
        input=probs, label=xe_label),
C
caoying03 已提交
43 44 45
    sum_cost(input=hidden),
    nce_layer(
        input=hidden, label=labels))