Hugging Face 核心模型库
Transformers 是 Hugging Face 维护的核心 Python 库,提供数千个预训练模型(BERT、GPT、Llama、T5 等)的统一接口。 一行代码就能加载模型做文本分类、生成、翻译、摘要——是 AI 开发的基础设施。
transformers 是 Hugging Face 最知名的 Python 库,支持 NLP、计算机视觉、音频 三大领域的预训练模型。
它把不同模型架构(BERT、GPT、Llama、ViT、Whisper……)统一成一套 API,你不需要懂每个模型的内部实现就能使用它。
yt-dlp、Strix 是应用工具——安装就能用。Transformers 是开发库——你要写 Python 代码调用它。
你需要 Python 基础,不需要深度学习知识也能开始。
# 核心库 + PyTorch(推荐)
pip install transformers torch
# 只装核心库(用 CPU 也能跑)
pip install transformers
# 验证安装
python -c "from transformers import pipeline; print('OK')"
Pipeline 是 transformers 的"一键式"接口——你指定任务名,它自动加载合适的模型和分词器:
from transformers import pipeline
# 创建情感分析 pipeline(一行代码)
分类器 = pipeline("sentiment-analysis")
# 直接使用
结果 = 分类器("今天天气真好啊")
print(结果)
# [{'label': 'POSITIVE', 'score': 0.998}]
| 任务名 | 作用 | 示例 |
|---|---|---|
"sentiment-analysis" | 情感分析 | "电影太棒了" → POSITIVE |
"text-generation" | 文本生成 | "从前有座山" → "山里有座庙" |
"summarization" | 摘要 | 长文 → 一句话概括 |
"translation" | 翻译 | "Hello" → "你好" |
"fill-mask" | 完形填空 | "我今天去<MASK>" → "上班" |
"question-answering" | 问答 | 根据文章回答问题 |
"zero-shot-classification" | 零样本分类 | 不用训练,直接分类到自定义类别 |
"ner" | 命名实体识别 | 识别出人名、地名、组织 |
"image-classification" | 图片分类 | "这张图是猫还是狗" |
"automatic-speech-recognition" | 语音识别 | 音频 → 文字 |
from transformers import pipeline
# ── 情感分析 ──
分类器 = pipeline("sentiment-analysis")
print(分类器("这个产品太好用了!"))
print(分类器("质量很差,不值这个价"))
# ── 文本生成 ──
生成器 = pipeline("text-generation", model="distilgpt2")
print(生成器("人工智能的未来是", max_length=30, num_return_sequences=2))
# ── 零样本分类 ──
零样本 = pipeline("zero-shot-classification")
结果 = 零样本(
"我想学一门编程语言,容易上手且能找到工作",
candidate_labels=["Python", "Java", "C++", "Rust"]
)
print(结果)
pip install transformers torch复制上面的"第一个 Pipeline"代码,保存为 test_pipeline.py 并运行。
把任务改成 "fill-mask",看看完形填空的效果。
下一课讲常用场景——文本分类、生成、翻译、摘要等任务的详细用法。