Hugging Face 的 transformers 库使用(持续完善)

Hugging Face 的 transformers 库是一个非常强大的工具,它为各种自然语言处理(NLP)任务提供了预训练的模型,并且简化了模型的加载、训练和推理过程。它支持多种预训练模型,如 BERT、GPT、T5、BART 等。

一、核心组件

  1. AutoTokenizer:用于文本的分词和编码;
  2. AutoModel:加载预训练模型的基础类;
  3. Trainer 和 TrainingArguments:用于微调模型的高阶工具;
  4. Pipeline:封装了从预处理到推理的完整流程,适合快速开发。

1.加载模型和分词器

tokenizer = AutoTokenizer.from_pretrained('bert-base-chinese')
model = AutoModel.from_pretrained( “google-bert/bert-base-cased” )

 from_pretrained()可以直接传入模型的名称,从huggingface中下载,也可以直接传入模型的本地路径

    评论
    添加红包

    请填写红包祝福语或标题

    红包个数最小为10个

    红包金额最低5元

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

    抵扣说明:

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

    余额充值