commit
fd4eeaf59c
@ -0,0 +1,6 @@
|
|||||||
|
data/raw_data
|
||||||
|
data/*.list
|
||||||
|
mnist_vgg_model
|
||||||
|
plot.png
|
||||||
|
train.log
|
||||||
|
*pyc
|
@ -0,0 +1,21 @@
|
|||||||
|
# Copyright (c) 2016 Baidu, Inc. All Rights Reserved
|
||||||
|
#
|
||||||
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
# you may not use this file except in compliance with the License.
|
||||||
|
# You may obtain a copy of the License at
|
||||||
|
#
|
||||||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
#
|
||||||
|
# Unless required by applicable law or agreed to in writing, software
|
||||||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
# See the License for the specific language governing permissions and
|
||||||
|
# limitations under the License.
|
||||||
|
|
||||||
|
o = open("./" + "train.list", "w")
|
||||||
|
o.write("./data/raw_data/train" +"\n")
|
||||||
|
o.close()
|
||||||
|
|
||||||
|
o = open("./" + "test.list", "w")
|
||||||
|
o.write("./data/raw_data/t10k" +"\n")
|
||||||
|
o.close()
|
@ -0,0 +1,22 @@
|
|||||||
|
#!/usr/bin/env sh
|
||||||
|
# This scripts downloads the mnist data and unzips it.
|
||||||
|
set -e
|
||||||
|
DIR="$( cd "$(dirname "$0")" ; pwd -P )"
|
||||||
|
rm -rf "$DIR/raw_data"
|
||||||
|
mkdir "$DIR/raw_data"
|
||||||
|
cd "$DIR/raw_data"
|
||||||
|
|
||||||
|
echo "Downloading..."
|
||||||
|
|
||||||
|
for fname in train-images-idx3-ubyte train-labels-idx1-ubyte t10k-images-idx3-ubyte t10k-labels-idx1-ubyte
|
||||||
|
do
|
||||||
|
if [ ! -e $fname ]; then
|
||||||
|
wget --no-check-certificate http://yann.lecun.com/exdb/mnist/${fname}.gz
|
||||||
|
gunzip ${fname}.gz
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
cd $DIR
|
||||||
|
rm -f *.list
|
||||||
|
python generate_list.py
|
||||||
|
|
@ -0,0 +1,32 @@
|
|||||||
|
from paddle.trainer.PyDataProvider2 import *
|
||||||
|
|
||||||
|
|
||||||
|
# Define a py data provider
|
||||||
|
@provider(input_types={
|
||||||
|
'pixel': dense_vector(28 * 28),
|
||||||
|
'label': integer_value(10)
|
||||||
|
})
|
||||||
|
def process(settings, filename): # settings is not used currently.
|
||||||
|
imgf = filename + "-images-idx3-ubyte"
|
||||||
|
labelf = filename + "-labels-idx1-ubyte"
|
||||||
|
f = open(imgf, "rb")
|
||||||
|
l = open(labelf, "rb")
|
||||||
|
|
||||||
|
f.read(16)
|
||||||
|
l.read(8)
|
||||||
|
|
||||||
|
# Define number of samples for train/test
|
||||||
|
if "train" in filename:
|
||||||
|
n = 60000
|
||||||
|
else:
|
||||||
|
n = 10000
|
||||||
|
|
||||||
|
for i in range(n):
|
||||||
|
label = ord(l.read(1))
|
||||||
|
pixels = []
|
||||||
|
for j in range(28 * 28):
|
||||||
|
pixels.append(float(ord(f.read(1))) / 255.0)
|
||||||
|
yield {"pixel": pixels, 'label': label}
|
||||||
|
|
||||||
|
f.close()
|
||||||
|
l.close()
|
@ -0,0 +1,31 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
# Copyright (c) 2016 Baidu, Inc. All Rights Reserved
|
||||||
|
#
|
||||||
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
# you may not use this file except in compliance with the License.
|
||||||
|
# You may obtain a copy of the License at
|
||||||
|
#
|
||||||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
#
|
||||||
|
# Unless required by applicable law or agreed to in writing, software
|
||||||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
# See the License for the specific language governing permissions and
|
||||||
|
# limitations under the License.
|
||||||
|
set -e
|
||||||
|
config=vgg_16_mnist.py
|
||||||
|
output=./mnist_vgg_model
|
||||||
|
log=train.log
|
||||||
|
|
||||||
|
paddle train \
|
||||||
|
--config=$config \
|
||||||
|
--dot_period=10 \
|
||||||
|
--log_period=100 \
|
||||||
|
--test_all_data_in_one_period=1 \
|
||||||
|
--use_gpu=0 \
|
||||||
|
--trainer_count=1 \
|
||||||
|
--num_passes=100 \
|
||||||
|
--save_dir=$output \
|
||||||
|
2>&1 | tee $log
|
||||||
|
|
||||||
|
python -m paddle.utils.plotcurve -i $log > plot.png
|
@ -0,0 +1,53 @@
|
|||||||
|
# Copyright (c) 2016 Baidu, Inc. All Rights Reserved
|
||||||
|
#
|
||||||
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
# you may not use this file except in compliance with the License.
|
||||||
|
# You may obtain a copy of the License at
|
||||||
|
#
|
||||||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
#
|
||||||
|
# Unless required by applicable law or agreed to in writing, software
|
||||||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
# See the License for the specific language governing permissions and
|
||||||
|
# limitations under the License.
|
||||||
|
|
||||||
|
from paddle.trainer_config_helpers import *
|
||||||
|
|
||||||
|
is_predict = get_config_arg("is_predict", bool, False)
|
||||||
|
|
||||||
|
####################Data Configuration ##################
|
||||||
|
|
||||||
|
|
||||||
|
if not is_predict:
|
||||||
|
data_dir='./data/'
|
||||||
|
define_py_data_sources2(train_list= data_dir + 'train.list',
|
||||||
|
test_list= data_dir + 'test.list',
|
||||||
|
module='mnist_provider',
|
||||||
|
obj='process')
|
||||||
|
|
||||||
|
######################Algorithm Configuration #############
|
||||||
|
settings(
|
||||||
|
batch_size = 128,
|
||||||
|
learning_rate = 0.1 / 128.0,
|
||||||
|
learning_method = MomentumOptimizer(0.9),
|
||||||
|
regularization = L2Regularization(0.0005 * 128)
|
||||||
|
)
|
||||||
|
|
||||||
|
#######################Network Configuration #############
|
||||||
|
|
||||||
|
data_size=1*28*28
|
||||||
|
label_size=10
|
||||||
|
img = data_layer(name='pixel', size=data_size)
|
||||||
|
|
||||||
|
# small_vgg is predined in trainer_config_helpers.network
|
||||||
|
predict = small_vgg(input_image=img,
|
||||||
|
num_channels=1,
|
||||||
|
num_classes=label_size)
|
||||||
|
|
||||||
|
if not is_predict:
|
||||||
|
lbl = data_layer(name="label", size=label_size)
|
||||||
|
inputs(img, lbl)
|
||||||
|
outputs(classification_cost(input=predict, label=lbl))
|
||||||
|
else:
|
||||||
|
outputs(predict)
|
@ -0,0 +1,62 @@
|
|||||||
|
# edit-mode: -*- python -*-
|
||||||
|
|
||||||
|
# Copyright (c) 2016 Baidu, Inc. All Rights Reserved
|
||||||
|
#
|
||||||
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
# you may not use this file except in compliance with the License.
|
||||||
|
# You may obtain a copy of the License at
|
||||||
|
#
|
||||||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
#
|
||||||
|
# Unless required by applicable law or agreed to in writing, software
|
||||||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
# See the License for the specific language governing permissions and
|
||||||
|
# limitations under the License.
|
||||||
|
|
||||||
|
from paddle.trainer_config_helpers import *
|
||||||
|
|
||||||
|
dict_file = "./data/dict.txt"
|
||||||
|
word_dict = dict()
|
||||||
|
with open(dict_file, 'r') as f:
|
||||||
|
for i, line in enumerate(f):
|
||||||
|
w = line.strip().split()[0]
|
||||||
|
word_dict[w] = i
|
||||||
|
|
||||||
|
is_predict = get_config_arg('is_predict', bool, False)
|
||||||
|
trn = 'data/train.list' if not is_predict else None
|
||||||
|
tst = 'data/test.list' if not is_predict else 'data/pred.list'
|
||||||
|
process = 'process' if not is_predict else 'process_predict'
|
||||||
|
define_py_data_sources2(train_list=trn,
|
||||||
|
test_list=tst,
|
||||||
|
module="dataprovider_emb",
|
||||||
|
obj=process,
|
||||||
|
args={"dictionary": word_dict})
|
||||||
|
|
||||||
|
batch_size = 128 if not is_predict else 1
|
||||||
|
settings(
|
||||||
|
batch_size=batch_size,
|
||||||
|
learning_rate=2e-3,
|
||||||
|
learning_method=AdamOptimizer(),
|
||||||
|
regularization=L2Regularization(8e-4),
|
||||||
|
gradient_clipping_threshold=25
|
||||||
|
)
|
||||||
|
|
||||||
|
bias_attr = ParamAttr(initial_std=0.,l2_rate=0.)
|
||||||
|
data = data_layer(name="word", size=len(word_dict))
|
||||||
|
emb = embedding_layer(input=data, size=128)
|
||||||
|
|
||||||
|
bi_lstm = bidirectional_lstm(input=emb, size=128)
|
||||||
|
dropout = dropout_layer(input=bi_lstm, dropout_rate=0.5)
|
||||||
|
|
||||||
|
output = fc_layer(input=dropout, size=2,
|
||||||
|
bias_attr=bias_attr,
|
||||||
|
act=SoftmaxActivation())
|
||||||
|
|
||||||
|
if is_predict:
|
||||||
|
maxid = maxid_layer(output)
|
||||||
|
outputs([maxid, output])
|
||||||
|
else:
|
||||||
|
label = data_layer(name="label", size=2)
|
||||||
|
cls = classification_cost(input=output, label=label)
|
||||||
|
outputs(cls)
|
@ -0,0 +1,73 @@
|
|||||||
|
# edit-mode: -*- python -*-
|
||||||
|
|
||||||
|
# Copyright (c) 2016 Baidu, Inc. All Rights Reserved
|
||||||
|
#
|
||||||
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
# you may not use this file except in compliance with the License.
|
||||||
|
# You may obtain a copy of the License at
|
||||||
|
#
|
||||||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
#
|
||||||
|
# Unless required by applicable law or agreed to in writing, software
|
||||||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
# See the License for the specific language governing permissions and
|
||||||
|
# limitations under the License.
|
||||||
|
|
||||||
|
from paddle.trainer_config_helpers import *
|
||||||
|
|
||||||
|
dict_file = "./data/dict.txt"
|
||||||
|
word_dict = dict()
|
||||||
|
with open(dict_file, 'r') as f:
|
||||||
|
for i, line in enumerate(f):
|
||||||
|
w = line.strip().split()[0]
|
||||||
|
word_dict[w] = i
|
||||||
|
|
||||||
|
is_predict = get_config_arg('is_predict', bool, False)
|
||||||
|
trn = 'data/train.list' if not is_predict else None
|
||||||
|
tst = 'data/test.list' if not is_predict else 'data/pred.list'
|
||||||
|
process = 'process' if not is_predict else 'process_predict'
|
||||||
|
define_py_data_sources2(train_list=trn,
|
||||||
|
test_list=tst,
|
||||||
|
module="dataprovider_emb",
|
||||||
|
obj=process,
|
||||||
|
args={"dictionary": word_dict})
|
||||||
|
|
||||||
|
batch_size = 128 if not is_predict else 1
|
||||||
|
settings(
|
||||||
|
batch_size=batch_size,
|
||||||
|
learning_rate=2e-3,
|
||||||
|
learning_method=AdamOptimizer(),
|
||||||
|
regularization=L2Regularization(8e-4),
|
||||||
|
gradient_clipping_threshold=25
|
||||||
|
)
|
||||||
|
|
||||||
|
bias_attr = ParamAttr(initial_std=0.,l2_rate=0.)
|
||||||
|
|
||||||
|
data = data_layer(name="word", size=len(word_dict))
|
||||||
|
emb = embedding_layer(input=data, size=128)
|
||||||
|
|
||||||
|
hidden_0 = mixed_layer(size=128, input=[full_matrix_projection(input=emb)])
|
||||||
|
lstm_0 = lstmemory(input=hidden_0, layer_attr=ExtraAttr(drop_rate=0.1))
|
||||||
|
|
||||||
|
input_layers = [hidden_0, lstm_0]
|
||||||
|
|
||||||
|
for i in range(1,8):
|
||||||
|
fc = fc_layer(input=input_layers, size=128)
|
||||||
|
lstm = lstmemory(input=fc, layer_attr=ExtraAttr(drop_rate=0.1),
|
||||||
|
reverse=(i % 2) == 1,)
|
||||||
|
input_layers = [fc, lstm]
|
||||||
|
|
||||||
|
lstm_last = pooling_layer(input=lstm, pooling_type=MaxPooling())
|
||||||
|
|
||||||
|
output = fc_layer(input=lstm_last, size=2,
|
||||||
|
bias_attr=bias_attr,
|
||||||
|
act=SoftmaxActivation())
|
||||||
|
|
||||||
|
if is_predict:
|
||||||
|
maxid = maxid_layer(output)
|
||||||
|
outputs([maxid, output])
|
||||||
|
else:
|
||||||
|
label = data_layer(name="label", size=2)
|
||||||
|
cls = classification_cost(input=output, label=label)
|
||||||
|
outputs(cls)
|
@ -0,0 +1,21 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
# Copyright (c) 2016 Baidu, Inc. All Rights Reserved
|
||||||
|
#
|
||||||
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
# you may not use this file except in compliance with the License.
|
||||||
|
# You may obtain a copy of the License at
|
||||||
|
#
|
||||||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
#
|
||||||
|
# Unless required by applicable law or agreed to in writing, software
|
||||||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
# See the License for the specific language governing permissions and
|
||||||
|
# limitations under the License.
|
||||||
|
set -e
|
||||||
|
|
||||||
|
DIR="$( cd "$(dirname "$0")" ; pwd -P )"
|
||||||
|
cd $DIR
|
||||||
|
|
||||||
|
wget http://www.cnts.ua.ac.be/conll2000/chunking/train.txt.gz
|
||||||
|
wget http://www.cnts.ua.ac.be/conll2000/chunking/test.txt.gz
|
@ -0,0 +1 @@
|
|||||||
|
data/test.txt.gz
|
@ -0,0 +1 @@
|
|||||||
|
data/train.txt.gz
|
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,84 @@
|
|||||||
|
# Copyright (c) 2016 Baidu, Inc. All Rights Reserved
|
||||||
|
#
|
||||||
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
# you may not use this file except in compliance with the License.
|
||||||
|
# You may obtain a copy of the License at
|
||||||
|
#
|
||||||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
#
|
||||||
|
# Unless required by applicable law or agreed to in writing, software
|
||||||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
# See the License for the specific language governing permissions and
|
||||||
|
# limitations under the License.
|
||||||
|
|
||||||
|
from paddle.trainer_config_helpers import *
|
||||||
|
|
||||||
|
import math
|
||||||
|
|
||||||
|
define_py_data_sources2(train_list="data/train.list",
|
||||||
|
test_list="data/test.list",
|
||||||
|
module="dataprovider",
|
||||||
|
obj="process")
|
||||||
|
|
||||||
|
|
||||||
|
batch_size = 1
|
||||||
|
settings(
|
||||||
|
learning_method=MomentumOptimizer(),
|
||||||
|
batch_size=batch_size,
|
||||||
|
regularization=L2Regularization(batch_size * 1e-4),
|
||||||
|
average_window=0.5,
|
||||||
|
learning_rate=1e-1,
|
||||||
|
learning_rate_decay_a=1e-5,
|
||||||
|
learning_rate_decay_b=0.25,
|
||||||
|
)
|
||||||
|
|
||||||
|
num_label_types=23
|
||||||
|
|
||||||
|
def get_simd_size(size):
|
||||||
|
return int(math.ceil(float(size) / 8)) * 8
|
||||||
|
|
||||||
|
# Currently, in order to use sparse_update=True,
|
||||||
|
# the size has to be aligned.
|
||||||
|
num_label_types = get_simd_size(num_label_types)
|
||||||
|
|
||||||
|
features = data_layer(name="features", size=76328)
|
||||||
|
word = data_layer(name="word", size=6778)
|
||||||
|
pos = data_layer(name="pos", size=44)
|
||||||
|
chunk = data_layer(name="chunk",
|
||||||
|
size=num_label_types)
|
||||||
|
|
||||||
|
crf_input = fc_layer(
|
||||||
|
input=features,
|
||||||
|
size=num_label_types,
|
||||||
|
act=LinearActivation(),
|
||||||
|
bias_attr=False,
|
||||||
|
param_attr=ParamAttr(initial_std=0, sparse_update=True))
|
||||||
|
|
||||||
|
crf=crf_layer(
|
||||||
|
input=crf_input,
|
||||||
|
label=chunk,
|
||||||
|
param_attr=ParamAttr(name="crfw", initial_std=0),
|
||||||
|
)
|
||||||
|
|
||||||
|
crf_decoding=crf_decoding_layer(
|
||||||
|
size=num_label_types,
|
||||||
|
input=crf_input,
|
||||||
|
label=chunk,
|
||||||
|
param_attr=ParamAttr(name="crfw"),
|
||||||
|
)
|
||||||
|
|
||||||
|
sum_evaluator(
|
||||||
|
name="error",
|
||||||
|
input=crf_decoding,
|
||||||
|
)
|
||||||
|
|
||||||
|
chunk_evaluator(
|
||||||
|
name="chunk_f1",
|
||||||
|
input =[crf_decoding, chunk],
|
||||||
|
chunk_scheme="IOB",
|
||||||
|
num_chunk_types=11,
|
||||||
|
)
|
||||||
|
|
||||||
|
inputs(word, pos, chunk, features)
|
||||||
|
outputs(crf)
|
@ -0,0 +1,45 @@
|
|||||||
|
# Sequence Tagging
|
||||||
|
|
||||||
|
This demo is a sequence model for assigning tags to each token in a sentence. The task is described at <a href = "http://www.cnts.ua.ac.be/conll2000/chunking">CONLL2000 Text Chunking</a> task.
|
||||||
|
|
||||||
|
## Download data
|
||||||
|
```bash
|
||||||
|
cd demo/sequence_tagging
|
||||||
|
./data/get_data.sh
|
||||||
|
```
|
||||||
|
|
||||||
|
## Train model
|
||||||
|
```bash
|
||||||
|
cd demo/sequence_tagging
|
||||||
|
./train.sh
|
||||||
|
```
|
||||||
|
|
||||||
|
## Model description
|
||||||
|
|
||||||
|
We provide two models. One is a linear CRF model (linear_crf.py) with is equivalent to the one at <a href="http://leon.bottou.org/projects/sgd#stochastic_gradient_crfs">leon.bottou.org/projects/sgd</a>. The second one is a stacked bidirectional RNN and CRF model (rnn_crf.py).
|
||||||
|
<center>
|
||||||
|
<table border="2" cellspacing="0" cellpadding="6" rules="all" frame="border">
|
||||||
|
|
||||||
|
<thead>
|
||||||
|
<th scope="col" class="left">Model name</th>
|
||||||
|
<th scope="col" class="left">Number of parameters</th>
|
||||||
|
<th scope="col" class="left">F1 score</th>
|
||||||
|
</thead>
|
||||||
|
|
||||||
|
<tbody>
|
||||||
|
<tr>
|
||||||
|
<td class="left">linear_crf</td>
|
||||||
|
<td class="left"> 1.8M </td>
|
||||||
|
<td class="left"> 0.937</td>
|
||||||
|
</tr>
|
||||||
|
|
||||||
|
<tr>
|
||||||
|
<td class="left">rnn_crf</td>
|
||||||
|
<td class="left"> 960K </td>
|
||||||
|
<td class="left">0.941</td>
|
||||||
|
</tr>
|
||||||
|
|
||||||
|
</tbody>
|
||||||
|
</table>
|
||||||
|
</center>
|
||||||
|
<br>
|
@ -0,0 +1,130 @@
|
|||||||
|
# Copyright (c) 2016 Baidu, Inc. All Rights Reserved
|
||||||
|
#
|
||||||
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
# you may not use this file except in compliance with the License.
|
||||||
|
# You may obtain a copy of the License at
|
||||||
|
#
|
||||||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
#
|
||||||
|
# Unless required by applicable law or agreed to in writing, software
|
||||||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
# See the License for the specific language governing permissions and
|
||||||
|
# limitations under the License.
|
||||||
|
|
||||||
|
from paddle.trainer_config_helpers import *
|
||||||
|
|
||||||
|
import math
|
||||||
|
|
||||||
|
define_py_data_sources2(train_list="data/train.list",
|
||||||
|
test_list="data/test.list",
|
||||||
|
module="dataprovider",
|
||||||
|
obj="process")
|
||||||
|
|
||||||
|
batch_size = 16
|
||||||
|
settings(
|
||||||
|
learning_method=MomentumOptimizer(),
|
||||||
|
batch_size=batch_size,
|
||||||
|
regularization=L2Regularization(batch_size * 1e-5),
|
||||||
|
average_window=0.5,
|
||||||
|
learning_rate = 2e-3,
|
||||||
|
learning_rate_decay_a = 5e-7,
|
||||||
|
learning_rate_decay_b = 0.5,
|
||||||
|
)
|
||||||
|
|
||||||
|
word_dim=128
|
||||||
|
hidden_dim = 128
|
||||||
|
with_rnn = True
|
||||||
|
|
||||||
|
initial_std=1/math.sqrt(hidden_dim)
|
||||||
|
param_attr=ParamAttr(initial_std=initial_std)
|
||||||
|
cpu_layer_attr=ExtraLayerAttribute(device=-1)
|
||||||
|
|
||||||
|
default_device(0)
|
||||||
|
|
||||||
|
num_label_types=23
|
||||||
|
|
||||||
|
features = data_layer(name="features", size=76328)
|
||||||
|
word = data_layer(name="word", size=6778)
|
||||||
|
pos = data_layer(name="pos", size=44)
|
||||||
|
chunk = data_layer(name="chunk",
|
||||||
|
size=num_label_types,
|
||||||
|
layer_attr=cpu_layer_attr)
|
||||||
|
|
||||||
|
emb = embedding_layer(
|
||||||
|
input=word, size=word_dim, param_attr=ParamAttr(initial_std=0))
|
||||||
|
|
||||||
|
hidden1 = mixed_layer(
|
||||||
|
size=hidden_dim,
|
||||||
|
act=STanhActivation(),
|
||||||
|
bias_attr=True,
|
||||||
|
input=[full_matrix_projection(emb),
|
||||||
|
table_projection(pos, param_attr=param_attr)]
|
||||||
|
)
|
||||||
|
|
||||||
|
if with_rnn:
|
||||||
|
rnn1 = recurrent_layer(
|
||||||
|
act=ReluActivation(),
|
||||||
|
bias_attr=True,
|
||||||
|
input=hidden1,
|
||||||
|
param_attr=ParamAttr(initial_std=0),
|
||||||
|
)
|
||||||
|
|
||||||
|
hidden2 = mixed_layer(
|
||||||
|
size=hidden_dim,
|
||||||
|
act=STanhActivation(),
|
||||||
|
bias_attr=True,
|
||||||
|
input=[full_matrix_projection(hidden1)
|
||||||
|
] + ([
|
||||||
|
full_matrix_projection(rnn1, param_attr=ParamAttr(initial_std=0))
|
||||||
|
] if with_rnn else []),
|
||||||
|
)
|
||||||
|
|
||||||
|
if with_rnn:
|
||||||
|
rnn2=recurrent_layer(
|
||||||
|
reverse=True,
|
||||||
|
act=ReluActivation(),
|
||||||
|
bias_attr=True,
|
||||||
|
input=hidden2,
|
||||||
|
param_attr=ParamAttr(initial_std=0),
|
||||||
|
)
|
||||||
|
|
||||||
|
crf_input = mixed_layer(
|
||||||
|
size=num_label_types,
|
||||||
|
bias_attr=False,
|
||||||
|
input=[
|
||||||
|
full_matrix_projection(hidden2),
|
||||||
|
] + ([
|
||||||
|
full_matrix_projection(rnn2, param_attr=ParamAttr(initial_std=0))
|
||||||
|
] if with_rnn else []),
|
||||||
|
)
|
||||||
|
|
||||||
|
crf = crf_layer(
|
||||||
|
input=crf_input,
|
||||||
|
label=chunk,
|
||||||
|
param_attr=ParamAttr(name="crfw", initial_std=0),
|
||||||
|
layer_attr=cpu_layer_attr,
|
||||||
|
)
|
||||||
|
|
||||||
|
crf_decoding = crf_decoding_layer(
|
||||||
|
size=num_label_types,
|
||||||
|
input=crf_input,
|
||||||
|
label=chunk,
|
||||||
|
param_attr=ParamAttr(name="crfw"),
|
||||||
|
layer_attr=cpu_layer_attr,
|
||||||
|
)
|
||||||
|
|
||||||
|
sum_evaluator(
|
||||||
|
name="error",
|
||||||
|
input=crf_decoding,
|
||||||
|
)
|
||||||
|
|
||||||
|
chunk_evaluator(
|
||||||
|
name="chunk_f1",
|
||||||
|
input =[crf_decoding, chunk],
|
||||||
|
chunk_scheme="IOB",
|
||||||
|
num_chunk_types=11,
|
||||||
|
)
|
||||||
|
|
||||||
|
inputs(word, pos, chunk, features)
|
||||||
|
outputs(crf)
|
@ -0,0 +1,10 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
paddle train \
|
||||||
|
--config rnn_crf.py \
|
||||||
|
--parallel_nn=1 \
|
||||||
|
--use_gpu=1 \
|
||||||
|
--dot_period=10 \
|
||||||
|
--log_period=1000 \
|
||||||
|
--test_period=0 \
|
||||||
|
--num_passes=10
|
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in new issue