未验证 提交 0ca2ab88 编写于 作者: S Steffy-zxf 提交者: GitHub

Update README.md

上级 40cfdab1
# PaddleHub 回归任务 # PaddleHub 回归任务
本示例将展示如何使用PaddleHub Finetune API以及BERT预训练模型完成分类任务。 本示例将展示如何使用PaddleHub Finetune API以及BERT预训练模型完成回归任务。
## 如何开始Finetune ## 如何开始Finetune
...@@ -91,7 +91,7 @@ config = hub.RunConfig(use_cuda=True, num_epoch=3, batch_size=32, strategy=strat ...@@ -91,7 +91,7 @@ config = hub.RunConfig(use_cuda=True, num_epoch=3, batch_size=32, strategy=strat
* `enable_memory_optim`: 是否使用内存优化, 默认为True * `enable_memory_optim`: 是否使用内存优化, 默认为True
* `strategy`: Finetune优化策略 * `strategy`: Finetune优化策略
### Step4: 构建网络并创建分类迁移任务进行Finetune ### Step4: 构建网络并创建回归迁移任务进行Finetune
```python ```python
pooled_output = outputs["pooled_output"] pooled_output = outputs["pooled_output"]
...@@ -112,8 +112,8 @@ reg_task = hub.RegressionTask( ...@@ -112,8 +112,8 @@ reg_task = hub.RegressionTask(
reg_task.finetune_and_eval() reg_task.finetune_and_eval()
``` ```
**NOTE:** **NOTE:**
1. `outputs["pooled_output"]`返回了ERNIE/BERT模型对应的[CLS]向量,可以用于句子或句对的特征表达。 1. `outputs["pooled_output"]`返回了BERT模型对应的[CLS]向量,可以用于句子或句对的特征表达。
2. `feed_list`中的inputs参数指名了ERNIE/BERT中的输入tensor的顺序,与ClassifyReader返回的结果一致。 2. `feed_list`中的inputs参数指名了BERT中的输入tensor的顺序,与RegressionReader返回的结果一致。
## 可视化 ## 可视化
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册