openAI,fine-tuning的示例代码

我们将微调一个ada分类器来区分两个运动:棒球和曲棍球

首先,你需要安装openai的Python包:

pip install openai
  • 先准备并处理数据
from sklearn.datasets import fetch_20newsgroups
import pandas as pd
import openai

categories = ['rec.sport.baseball', 'rec.sport.hockey']
sports_dataset = fetch_20newsgroups(subset='train', shuffle=True, random_state=42, categories=categories)
labels = [sports_dataset.target_names[x].split('.')[-1] for x in sports_dataset['target']]
texts = [text.strip() for text in sports_dataset['data']]
df = pd.DataFrame(zip(texts, labels), columns = ['prompt','completion']) #[:300]
df.head()
df.to_json("sport2.jsonl", orient='records', lines=True)
  • 用openai的工具进行数据预处理
!openai tools fine_tunes.prepare_data -f sport2.jsonl -q

我们运行以下命令来训练数据集。由于这是一个分类任务,我们想知道在提供的验证集上的泛化性能,以用于我们的分类应用场景。该工具建议添加–compute_classification_metrics --classification_positive_class "baseball"以计算分类指标。
我们可以直接从CLI工具复制建议的命令。我们特别添加了-m ada来微调一个更便宜和更快的ada模型,通常在分类应用场景中与更慢更昂贵的模型相当。

!openai api fine_tunes.create -t "sport2_prepared_train.jsonl" -v "sport2_prepared_valid.jsonl" --compute_classification_metrics --classification_positive_class " baseball" -m ada
  • 我们现在可以下载结果文件,观察在保留的验证集上的预期性能。
!openai api fine_tunes.results -i ft-2zaA7qi0rxJduWQpdvOvmGn3 > result.csv
  • 用代码查看训练结果
results = pd.read_csv('result.csv')
results[results['classification/accuracy'].notnull()].tail(1)
results[results['classification/accuracy'].notnull()]['classification/accuracy'].plot()
  • 调用模型,加载校验数据
test = pd.read_json('sport2_prepared_valid.jsonl', lines=True)
test.head()

我们需要使用与微调期间相同的分隔符,即\n\n###\n\n。在这种情况下,由于我们关心分类,因此我们希望温度尽可能低,并且只需要一个标记完成来确定模型的预测。

ft_model = 'ada:ft-openai-2021-07-30-12-26-20'
res = openai.Completion.create(model=ft_model, prompt=test['prompt'][0] + '\n\n###\n\n', max_tokens=1, temperature=0)
res['choices'][0]['text']

为了获取对数概率,我们可以在完成请求上指定logprobs参数。

res = openai.Completion.create(model=ft_model, prompt=test['prompt'][0] + '\n\n###\n\n', max_tokens=1, temperature=0, logprobs=2)
#查看结果
print(res['choices'][0]['logprobs']['top_logprobs'][0])

过请求log_probs,我们可以看到每个类别的预测(对数)概率。

向量数据库简介
RedisTimeSeries开源的时序数据数据库
BNF 语法描述
python将抽象语法树转换回源代码的工具库astor
Python 的抽象语法树库ast
python可以执行字符串形式的 Python 代码的库exec
python用于解析复杂文本数据的库Ply
python 用于解析复杂文本数据的库PyParsing
python用来进行代码语法高亮的库Pygments
Pylint
python处理网格数据的一个库GridDataFormats
python开发的开源数学软件系统SageMath
Python端到端的测试的生态系统库pyATS
Python 强大的模板引擎库 Skeleton BootStrap
python读取和编写配置文件库ConfigObj和ConfigParser
python在Web应用程序中安全地转义和渲染HTML的库MarkupSafe
Python为命令行界面(CLI)工具自动生成解析器的库Docopt
python的模板引擎库Mako,生成代码也很简单
python生成PDF文档的库reportlab
python的生成艺术字体的库pythonwordart

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

openwin_top

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值