上云无忧 > 文档中心 > 百度智能云文心大模型API ERNIE 3.0模型精调 - 快速开始
文心大模型API
百度智能云文心大模型API ERNIE 3.0模型精调 - 快速开始

文档简介:
邀测申请: 目前ERNIE 3.0系列的模型精调处于邀测阶段,您可以通过以下方式申请加入试用: 申请表单:https://wenxin.baidu.com/wenxin/apply3 【百度智能云】文心一言
*此产品及展示信息均由百度智能云官方提供。免费试用 咨询热线:400-826-7010,为您提供专业的售前咨询,让您快速了解云产品,助您轻松上云! 微信咨询
  免费试用、价格特惠

邀测申请

目前ERNIE 3.0系列的模型精调处于邀测阶段,您可以通过以下方式申请加入试用:

申请表单:https://wenxin.baidu.com/wenxin/apply3

模型精调基本步骤

获取API Key

您可以登录大模型API服务平台个人中心获取您的专属 API Key(AK)和Secret Key(SK),请注意保护您的密钥信息,避免泄露。您可以通过删除已泄露的密钥来保护您的账户安全。

安装wenxin-api客户端

安装文心大模型API客户端wenxin-api

pip install --upgrade wenxin-api

(注意:建议使用python3.7及以上版本)

模型精调接口调用

您可以通过以下代码使用您的数据集进行模型精调(将上一步骤申请的AK、SK填入下方代码的your ak和your sk中)。

# -*- coding: utf-8 -* import time import wenxin_api from wenxin_api
 import Dataset, Model, Train, Task from wenxin_api.const import TASK_STATE_TRAIN_SUCCESS

wenxin_api.ak = "your ak" wenxin_api.sk = "your sk" # 上传用户自定义数据集 
local_file_path = "local_file_path" dataset = Dataset.create(local_file_path) # 
创建训练任务 # 注意请求频率,过于频繁的请求会被限制qps sleep_time = 20 train_task 
= Train.create(train_datasets=[dataset]) while train_task.status != TASK_STATE
_TRAIN_SUCCESS: train_task.update() print("task:", train_task) time.sleep(sleep
_time) model = Model.retrieve(task_id=train_task.id) input_text = "标题:小熊过桥。
大纲:小熊成功走过|小熊害怕|小熊送给|小熊想|看望姥姥|妈妈道别|带上鲜花|竹桥走。
正文:" rst = Task.create(model=model, text=input_text)

其中常见的状态码可以参考:状态码汇总

相似文档
官方微信
联系客服
400-826-7010
7x24小时客服热线
分享
  • QQ好友
  • QQ空间
  • 微信
  • 微博
返回顶部