提交 c9f6394f 编写于 作者: W wuzewu 提交者: bbking

Update emotion detection README (#2476)

上级 3c287b42
...@@ -168,6 +168,8 @@ python tokenizer.py --test_data_dir ./test.txt.utf8 --batch_size 1 > test.txt.ut ...@@ -168,6 +168,8 @@ python tokenizer.py --test_data_dir ./test.txt.utf8 --batch_size 1 > test.txt.ut
--init_checkpoint ./models/ernie_finetune/params --init_checkpoint ./models/ernie_finetune/params
``` ```
#### 如何基于PaddleHub加载ERNIE进行 Finetune
我们也提供了使用PaddleHub加载ERNIE模型的选项,PaddleHub是PaddlePaddle的预训练模型管理工具,可以一行代码完成预训练模型的加载,简化预训练模型的使用和迁移学习。更多相关的介绍,可以查看[PaddleHub](https://github.com/PaddlePaddle/PaddleHub) 我们也提供了使用PaddleHub加载ERNIE模型的选项,PaddleHub是PaddlePaddle的预训练模型管理工具,可以一行代码完成预训练模型的加载,简化预训练模型的使用和迁移学习。更多相关的介绍,可以查看[PaddleHub](https://github.com/PaddlePaddle/PaddleHub)
如果想使用该功能,需要修改run_ernie.sh中的配置如下: 如果想使用该功能,需要修改run_ernie.sh中的配置如下:
...@@ -178,7 +180,24 @@ python tokenizer.py --test_data_dir ./test.txt.utf8 --batch_size 1 > test.txt.ut ...@@ -178,7 +180,24 @@ python tokenizer.py --test_data_dir ./test.txt.utf8 --batch_size 1 > test.txt.ut
注意:使用该选项需要先安装PaddleHub,安装命令如下 注意:使用该选项需要先安装PaddleHub,安装命令如下
```shell ```shell
$ pip install paddlehub pip install paddlehub
```
执行以下命令进行Finetune
```shell
sh run_ernie.sh train
```
Finetune结束后,进行eval或者infer时,需要修改run_ernie.sh中的配置如下:
```shell
# 在eval()和infer()函数中,修改--use_paddle_hub选项
--use_paddle_hub true
```
执行以下命令进行eval和infer
```shell
sh run_ernie.sh eval
sh run_ernie.sh infer
``` ```
## 如何贡献代码 ## 如何贡献代码
......
...@@ -45,7 +45,7 @@ data_g.add_arg("data_dir", str, None, "Directory path to training data.") ...@@ -45,7 +45,7 @@ data_g.add_arg("data_dir", str, None, "Directory path to training data.")
data_g.add_arg("vocab_path", str, None, "Vocabulary path.") data_g.add_arg("vocab_path", str, None, "Vocabulary path.")
data_g.add_arg("batch_size", int, 256, "Total examples' number in batch for training.") data_g.add_arg("batch_size", int, 256, "Total examples' number in batch for training.")
data_g.add_arg("random_seed", int, 0, "Random seed.") data_g.add_arg("random_seed", int, 0, "Random seed.")
data_g.add_arg("num_labels", int, 2, "label number") data_g.add_arg("num_labels", int, 3, "label number")
data_g.add_arg("max_seq_len", int, 512, "Number of words of the longest seqence.") data_g.add_arg("max_seq_len", int, 512, "Number of words of the longest seqence.")
data_g.add_arg("train_set", str, None, "Path to training data.") data_g.add_arg("train_set", str, None, "Path to training data.")
data_g.add_arg("test_set", str, None, "Path to test data.") data_g.add_arg("test_set", str, None, "Path to test data.")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册