diff --git a/fluid/PaddleNLP/chinese_ner/README.md b/fluid/PaddleNLP/chinese_ner/README.md index c5155b181b45eddcab3cd8d02d2036bd8c1e93ad..a458c83b5f1ad9c007d35ddfb7a6578fb14bbf2a 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 7684f683e77b35be84b5753793f97308c7763cd8..fc65528cd34706ea905025702cbea0307bef0686 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,