未验证 提交 79aef5c8 编写于 作者: Y Yibing Liu 提交者: GitHub

Fix readme in chinese ner (#1684)

上级 8908f2e3
...@@ -15,7 +15,14 @@ ...@@ -15,7 +15,14 @@
在data目录下,有两个文件夹,train_files中保存的是训练数据,test_files中保存的是测试数据,作为示例,在目录下我们各放置了两个文件,实际训练时,根据自己的实际需要将数据放置在对应目录,并根据数据格式,修改reader.py中的数据读取函数。 在data目录下,有两个文件夹,train_files中保存的是训练数据,test_files中保存的是测试数据,作为示例,在目录下我们各放置了两个文件,实际训练时,根据自己的实际需要将数据放置在对应目录,并根据数据格式,修改reader.py中的数据读取函数。
## 训练 ## 训练
修改 [train.py](./train.py)`main` 函数,指定数据路径,运行`python train.py`开始训练。
通过运行
```
python train.py --help
```
来获取命令行参数的帮助,设置正确的数据路径等参数后,运行`train.py`开始训练。
训练记录形如 训练记录形如
```txt ```txt
...@@ -31,7 +38,7 @@ pass_id:2, time_cost:0.740842103958s ...@@ -31,7 +38,7 @@ pass_id:2, time_cost:0.740842103958s
``` ```
## 预测 ## 预测
修改 [infer.py](./infer.py)`infer` 函数,指定:需要测试的模型的路径、测试数据、预测标记文件的路径,运行`python infer.py`开始预测。 类似于训练过程,预测时指定需要测试模型的路径、测试数据、预测标记文件的路径,运行`infer.py`开始预测。
预测结果如下 预测结果如下
```txt ```txt
......
...@@ -12,7 +12,7 @@ import reader ...@@ -12,7 +12,7 @@ import reader
def parse_args(): def parse_args():
parser = argparse.ArgumentParser("Run inference.") parser = argparse.ArgumentParser("Run training.")
parser.add_argument( parser.add_argument(
'--batch_size', '--batch_size',
type=int, type=int,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册