Hugging face学习笔记

关于使用pipeline,方便,就是task参数有点难记,可以参考下面的文档

https://huggingface.co/docs/transformers/v4.36.1/zh/quicktour

作为个人学习《hugging face自然语言处理详解》一书的笔记

一.准备数据集

1.加载编码工具

2.准备数据集

准备好数据集和编码工具后,要用到dataset.map和dataset.filter函数,dataset.map函数用于给数据编码,filter进行进一步的处理,过滤掉一些无法输入模型的样本。要向这两个函数里面输入构造的函数。

二.定义模型与训练工具

1.加载预训练模型

2.定义评价函数 compute_metrics()根据不同任务自行定义

3.定义训练超参数 TrainingArguments

4.定义训练器 Trainer

数据整理函数

三.训练和测试

1.trainer.evaluate

2.trainer.train

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值