提交 200fcf42 编写于 作者: Z zhangwenhui03

fix bug

上级 0a268ea1
...@@ -79,7 +79,7 @@ SessionId ItemId Time ...@@ -79,7 +79,7 @@ SessionId ItemId Time
2 214757407 1396850438.247 2 214757407 1396850438.247
``` ```
数据格式需要转换 运行脚本 数据格式需要转换, 运行脚本如下
``` ```
python convert_format.py python convert_format.py
``` ```
...@@ -101,7 +101,7 @@ python convert_format.py ...@@ -101,7 +101,7 @@ python convert_format.py
根据训练和测试文件生成字典和对应的paddle输入文件 根据训练和测试文件生成字典和对应的paddle输入文件
注意需要将训练文件放到一个目录下面,测试文件放到一个目录下面,同时支持多训练文件 需要将训练文件放到目录raw_train_data下,测试文件放到目录raw_test_data下,并生成对应的train_data,test_data和vocab.txt文件
``` ```
python text2paddle.py raw_train_data/ raw_test_data/ train_data test_data vocab.txt python text2paddle.py raw_train_data/ raw_test_data/ train_data test_data vocab.txt
``` ```
......
...@@ -81,7 +81,7 @@ def infer(args, vocab_size, test_reader): ...@@ -81,7 +81,7 @@ def infer(args, vocab_size, test_reader):
start_up_program = fluid.Program() start_up_program = fluid.Program()
with fluid.program_guard(main_program, start_up_program): with fluid.program_guard(main_program, start_up_program):
acc = model(vocab_size, emb_size, hid_size) acc = model(vocab_size, emb_size, hid_size)
for epoch in xrange(start_index, last_index + 1): for epoch in range(start_index, last_index + 1):
copy_program = main_program.clone() copy_program = main_program.clone()
model_path = model_dir + "/epoch_" + str(epoch) model_path = model_dir + "/epoch_" + str(epoch)
fluid.io.load_params( fluid.io.load_params(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册