From 79aef5c8117ab5c68944c7047f26144e183fa05a Mon Sep 17 00:00:00 2001 From: Yibing Liu Date: Wed, 23 Jan 2019 15:27:01 +0800 Subject: [PATCH] Fix readme in chinese ner (#1684) --- fluid/PaddleNLP/chinese_ner/README.md | 11 +++++++++-- fluid/PaddleNLP/chinese_ner/train.py | 2 +- 2 files changed, 10 insertions(+), 3 deletions(-) diff --git a/fluid/PaddleNLP/chinese_ner/README.md b/fluid/PaddleNLP/chinese_ner/README.md index c5155b18..a458c83b 100644 --- a/fluid/PaddleNLP/chinese_ner/README.md +++ b/fluid/PaddleNLP/chinese_ner/README.md @@ -15,7 +15,14 @@ 在data目录下,有两个文件夹,train_files中保存的是训练数据,test_files中保存的是测试数据,作为示例,在目录下我们各放置了两个文件,实际训练时,根据自己的实际需要将数据放置在对应目录,并根据数据格式,修改reader.py中的数据读取函数。 ## 训练 -修改 [train.py](./train.py) 的 `main` 函数,指定数据路径,运行`python train.py`开始训练。 + +通过运行 + +``` +python train.py --help +``` + +来获取命令行参数的帮助,设置正确的数据路径等参数后,运行`train.py`开始训练。 训练记录形如 ```txt @@ -31,7 +38,7 @@ pass_id:2, time_cost:0.740842103958s ``` ## 预测 -修改 [infer.py](./infer.py) 的 `infer` 函数,指定:需要测试的模型的路径、测试数据、预测标记文件的路径,运行`python infer.py`开始预测。 +类似于训练过程,预测时指定需要测试模型的路径、测试数据、预测标记文件的路径,运行`infer.py`开始预测。 预测结果如下 ```txt diff --git a/fluid/PaddleNLP/chinese_ner/train.py b/fluid/PaddleNLP/chinese_ner/train.py index 7684f683..fc65528c 100644 --- a/fluid/PaddleNLP/chinese_ner/train.py +++ b/fluid/PaddleNLP/chinese_ner/train.py @@ -12,7 +12,7 @@ import reader def parse_args(): - parser = argparse.ArgumentParser("Run inference.") + parser = argparse.ArgumentParser("Run training.") parser.add_argument( '--batch_size', type=int, -- GitLab