公司微网站建设方案,coding.net wordpress,宁德市路桥建设有限公司网站,263企业邮箱下载客户端1. 概述
自然语言处理#xff08;Natural Language Processing#xff0c;NLP#xff09;是一门借助计算机技术研究人类语言的科学#xff0c;是人工智能领域的一个分支#xff0c;旨在让计算机理解、生成和处理人类语言。其核心任务是将非结构化的自然语言转换为机器可以…1. 概述
自然语言处理Natural Language ProcessingNLP是一门借助计算机技术研究人类语言的科学是人工智能领域的一个分支旨在让计算机理解、生成和处理人类语言。其核心任务是将非结构化的自然语言转换为机器可以理解和处理的形式并生成符合语义和语法的语言输出。虽然该领域的发展历史不长但是其发展迅速并且取得了许多令人印象深刻的成果。
2017 年 Google 在《Attention Is All You Need》中提出了 Transformer 结构用于序列标注在翻译任务上超过了之前最优秀的循环神经网络模型与此同时Fast AI 在《Universal Language Model Fine-tuning for Text Classification》中提出了一种名为 ULMFiT 的迁移学习方法将在大规模数据上预训练好的 LSTM 模型迁移用于文本分类只用很少的标注数据就达到了最佳性能。这些具有开创性的工作促成了两个著名 Transformer 模型的出现
GPT (the Generative Pretrained Transformer)BERT (Bidirectional Encoder Representations from Transformers)。
通过将 Transformer 结构与无监督学习相结合我们不再需要对每一个任务都从头开始训练模型并且几乎在所有 NLP 任务上都远远超过先前的最强基准。GPT 和 BERT 被提出之后NLP 领域出现了越来越多基于 Transformer 结构的模型。虽然新的 Transformer 模型层出不穷它们采用不同的预训练目标在不同的数据集上进行训练但是依然可以按模型结构将它们大致分为三类 纯 Encoder 模型又称自编码 (auto-encoding) Transformer 模型例如 BERT适用于只需要理解输入语义的任务例如句子分类、命名实体识别。在每个阶段注意力层都可以访问到原始输入句子中的所有词语即具有“双向 (Bi-directional)”注意力。 纯 Encoder 模型通常通过破坏给定的句子例如随机遮盖其中的词语然后让模型进行重构来进行预训练最适合处理那些需要理解整个句子语义的任务。 纯 Decoder 模型在每个阶段对于给定的词语注意力层只能访问句子中位于它之前的词语即只能迭代地基于已经生成的词语来逐个预测后面的词语因此也被称为自回归 (auto-regressive) 模型例如 GPT。适用于生成式任务例如文本生成。 Encoder-Decoder 模型又称 Seq2Seq (sequence-to-sequence) Transformer 模型例如 BART、T5适用于需要基于输入的生成式任务例如翻译、摘要。在每个阶段Encoder 的注意力层都可以访问初始输入句子中的所有单词而 Decoder 的注意力层则只能访问输入中给定词语之前的词语即已经解码生成的词语。 2. 原理
2.1 分词器
因为神经网络模型不能直接处理文本因此我们需要先将文本转换为数字这个过程被称为编码 (Encoding)其包含两个步骤
使用分词器 (tokenizer) 将文本按词、子词、字符切分为 tokens将所有的 token 映射到对应的 token ID。
词表就是一个映射字典负责将 token 映射到对应的 ID从 0 开始。神经网络模型就是通过这些 token ID 来区分每一个 token。
根据切分粒度的不同分词策略可以分为以下几种
1按词切分 (Word-based) 这种策略问题是会将文本中所有出现过的独立片段都作为不同的 token从而产生巨大的词表。而实际上很多词是相关的例如 “dog” 和 “dogs”、“run” 和 “running”如果给它们赋予不同的编号就无法表示出这种关联性。
当遇到不在词表中的词时分词器会使用一个专门的 [UNK] token 来表示它是 unknown 的。显然如果分词结果中包含很多 就意味着丢失了很多文本信息因此一个好的分词策略应该尽可能不出现 unknown token。
2按字符切分 (Character-based) 这种策略把文本切分为字符而不是词语这样就只会产生一个非常小的词表并且很少会出现词表外的 tokens。
但是从直觉上来看字符本身并没有太大的意义因此将文本切分为字符之后就会变得不容易理解。这也与语言有关例如中文字符会比拉丁字符包含更多的信息相对影响较小。此外这种方式切分出的 tokens 会很多例如一个由 10 个字符组成的单词就会输出 10 个 tokens而实际上它们只是一个词。
3按子词切分 (Subword) 高频词直接保留低频词被切分为更有意义的子词。例如 “annoyingly” 是一个低频词可以切分为 “annoying” 和 “ly”这两个子词不仅出现频率更高而且词义也得以保留。下图展示了对 “Let’s do tokenization!“ 按子词切分的结果 这种策略只用一个较小的词表就可以覆盖绝大部分文本基本不会产生 unknown token。“tokenization” 被切分为了 “token” 和 “ization”不仅保留了语义而且只用两个 token 就表示了一个长词。因此现在广泛采用的是一种同时结合了按词切分和按字符切分的方式——按子词切分 (Subword tokenization)。
2.2 编码与解码
1文本编码
分词使用分词器按某种策略将文本切分为 tokens。映射将 tokens 转化为对应的 token IDs。
rom transformers import AutoTokenizertokenizer AutoTokenizer.from_pretrained(bert-base-cased)sequence Using a Transformer network is simple
tokens tokenizer.tokenize(sequence)
print(tokens)ids tokenizer.convert_tokens_to_ids(tokens)
print(ids)[using, a, transform, ##er, network, is, simple]
[7993, 170, 13809, 23763, 2443, 1110, 3014]还可以通过 encode() 函数将这两个步骤合并并且 encode() 会自动添加模型需要的特殊 token例如 BERT 分词器会分别在序列的首尾添加 [CLS] 和 [SEP]。
from transformers import AutoTokenizertokenizer AutoTokenizer.from_pretrained(bert-base-cased)sequence Using a Transformer network is simple
sequence_ids tokenizer.encode(sequence)print(sequence_ids)[101, 7993, 170, 13809, 23763, 2443, 1110, 3014, 102]其中 101 和 102 分别是特殊 token 对应的 token IDs。
在实际编码文本时最常见的是直接使用分词器进行处理这样不仅会返回分词后的 token IDs还包含模型需要的其他输入。
tokenizer AutoTokenizer.from_pretrained(bert-base-cased)
tokenized_text tokenizer(Using a Transformer network is simple)2文本解码 文本解码 (Decoding) 与编码相反负责将 token IDs 转换回原来的字符串。注意解码过程不是简单地将 token IDs 映射回 tokens还需要合并那些被分为多个 token 的单词。
decoded_string tokenizer.decode([7993, 170, 11303, 1200, 2443, 1110, 3014])
print(decoded_string)3. 经典模型
3.1 NNLM 模型
2003 年本吉奥Bengio提出了神经网络语言模型Neural Network Language ModelNNLM)。可惜它生不逢时由于神经网络在当时并不被人们看好在之后的十年中 NNLM 模型都没有引起很大关注。直到 2013 年随着越来越多的研究者使用深度学习模型来处理自然语言NNLM 模型才被重新发掘并成为使用神经网络建模语言的经典范例。NNLM 模型的思路与统计语言模型保持一致它通过输入词语前面的 n-1 个词语来预测当前词。
具体来说NNLM 模型首先从词表 C 中查询得到前面 n-1 个词语对应的词向量 然后将这些词向量拼接后输入到带有激活函数的隐藏层中通过 Softmax 函数预测当前词语的概率。特别地包含所有词向量的词表矩阵 C 也是模型的参数需要通过学习获得。因此 NNLM 模型不仅能够能够根据上文预测当前词语同时还能够给出所有词语的词向量Word Embedding。
3.2 Word2Vec 模型
真正将神经网络语言模型发扬光大的是 2013 年 Google 公司提出的 Word2Vec 模型。Word2Vec 模型提供的词向量在很长一段时间里都是自然语言处理方法的标配即使是后来出现的 Glove 模型也难掩它的光芒。Word2Vec 的模型结构和 NNLM 基本一致只是训练方法有所不同分为 CBOW (Continuous Bag-of-Words) 和 Skip-gram 两种。 其中 CBOW 使用周围的词语 w ( t − 2 ) , w ( t − 1 ) , w ( t 1 ) , w ( t 2 ) w(t-2),w(t-1),w(t1),w(t2) w(t−2),w(t−1),w(t1),w(t2) 来预测当前词 w ( t ) w(t) w(t) 而 Skip-gram 则正好相反它使用当前词 w ( t ) w(t) w(t) 来预测它的周围词语。
可以看到与严格按照统计语言模型结构设计的 NNLM 模型不同Word2Vec 模型在结构上更加自由训练目标也更多地是为获得词向量服务。特别是同时通过上文和下文来预测当前词语的 CBOW 训练方法打破了语言模型“只通过上文来预测当前词”的固定思维为后续一系列神经网络语言模型的发展奠定了基础。
然而有一片乌云一直笼罩在 Word2Vec 模型的上空——多义词问题。一词多义是语言灵活性和高效性的体现但是 Word2Vec 模型却无法处理多义词一个词语无论表达何种语义Word2Vec 模型都只能提供相同的词向量即将多义词编码到了完全相同的参数空间。
实际上在 20 世纪 90 年代初雅让斯基Yarowsky就给出了一个简洁有效的解决方案——运用词语之间的互信息Mutual Information。具体来说对于多义词可以使用文本中与其同时出现的互信息最大的词语集合来表示不同的语义。例如对于“苹果”当表示水果时周围出现的一般就是“超市”、“香蕉”等词语而表示“苹果公司”时周围出现的一般就是“手机”、“平板”等词语。因此在判断多义词究竟表达何种语义时只需要查看哪个语义对应集合中的词语在上下文中出现的更多就可以了即通过上下文来判断语义。
3.3 ELMo 模型
为了更好地解决多义词问题2018 年研究者提出了 ELMo 模型Embeddings from Language Models。与 Word2Vec 模型只能提供静态词向量不同ELMo 模型会根据上下文动态地调整词语的词向量。
具体来说ELMo 模型首先对语言模型进行预训练使得模型掌握编码文本的能力然后在实际使用时对于输入文本中的每一个词语都提取模型各层中对应的词向量拼接起来作为新的词向量。ELMo 模型采用双层双向 LSTM 作为编码器从两个方向编码词语的上下文信息相当于将编码层直接封装到了语言模型中。 训练完成后 ELMo 模型不仅学习到了词向量还训练好了一个双层双向的 LSTM 编码器。对于输入文本中的词语可以从第一层 LSTM 中得到包含句法信息的词向量从第二层 LSTM 中得到包含语义信息的词向量最终通过加权求和得到每一个词语最终的词向量。
但是 ELMo 模型存在两个缺陷首先它使用 LSTM 模型作为编码器而不是当时已经提出的编码能力更强的 Transformer 模型其次 ELMo 模型直接通过拼接来融合双向抽取特征的做法也略显粗糙。
不久之后将 ELMo 模型中的 LSTM 更换为 Transformer 的 GPT 模型就出现了。但是 GPT 模型再次追随了 NNLM 的脚步只通过词语的上文进行预测这在很大程度上限制了模型的应用场景。例如对于文本分类、阅读理解等任务如果不把词语的下文信息也嵌入到词向量中就会白白丢掉很多信息。
3.4 BERT 模型
2018 年底随着 BERT 模型Bidirectional Encoder Representations from Transformers的出现这一阶段神经网络语言模型的发展终于出现了一位集大成者发布时 BERT 模型在 11 个任务上都取得了最好性能。
BERT 模型采用和 GPT 模型类似的两阶段框架首先对语言模型进行预训练然后通过微调来完成下游任务。但是BERT 不仅像 GPT 模型一样采用 Transformer 作为编码器而且采用了类似 ELMo 模型的双向语言模型结构。因此 BERT 模型不仅编码能力强大而且对各种下游任务BERT 模型都可以通过简单地改造输出部分来完成。 但是 BERT 模型的优点同样也是它的缺陷由于 BERT 模型采用双向语言模型结构因而无法直接用于生成文本。
可以看到从 2003 年 NNLM 模型提出时的无人问津到 2018 年底 BERT 模型横扫自然语言处理领域神经网络语言模型的发展也经历了一波三折。在此期间研究者一直在不断地对前人的工作进行改进这才取得了 BERT 模型的成功。BERT 模型的出现并非一蹴而就它不仅借鉴了 ELMo、GPT 等模型的结构与框架而且延续了 Word2Vec 模型提出的 CBOW 训练方式的思想可以看作是这一阶段语言模型发展的集大成者。
3.5 大语言模型
除了优化模型结构研究者发现扩大模型规模也可以提高性能。在保持模型结构以及预训练任务基本不变的情况下仅仅通过扩大模型规模就可以显著增强模型能力尤其当规模达到一定程度时模型甚至展现出了能够解决未见过复杂问题的涌现Emergent Abilities能力。例如 175B 规模的 GPT-3 模型只需要在输入中给出几个示例就能通过上下文学习In-context Learning完成各种小样本Few-Shot任务而这是 1.5B 规模的 GPT-2 模型无法做到的。
在规模扩展定律Scaling Laws被证明对语言模型有效之后研究者基于 Transformer 结构不断加深模型深度构建出了许多大语言模型。
一个标志性的事件是 2022 年 11 月 30 日 OpenAI 公司发布了面向普通消费者的 ChatGPT 模型Chat Generative Pre-trained Transformer它能够记住先前的聊天内容真正像人类一样交流甚至能撰写诗歌、论文、文案、代码等。发布后ChatGPT 模型引起了巨大轰动上线短短 5 天注册用户数就超过 100 万。2023 年一月末ChatGPT 活跃用户数量已经突破 1 亿成为史上增长最快的消费者应用。
4. 推理部署
4.1 步骤
将文本预处理为模型可以理解的格式。将预处理好的文本送入模型。对模型的预测值进行后处理输出人类可以理解的格式。 1使用分词器进行预处理
神经网络模型无法直接处理文本因此首先需要通过预处理环节将文本转换为模型可以理解的数字。具体地我们会使用每个模型对应的分词器 (tokenizer) 来进行
将输入切分为词语、子词或者符号例如标点符号统称为 tokens根据模型的词表将每个 token 映射到对应的 token 编号就是一个数字根据模型的需要添加一些额外的输入。
我们对输入文本的预处理需要与模型自身预训练时的操作完全一致只有这样模型才可以正常地工作。注意每个模型都有特定的预处理操作如果对要使用的模型不熟悉可以通过 Model Hub 查询。这里我们使用 AutoTokenizer 类和它的 from_pretrained() 函数它可以根据指定的模型 checkpoint 名称自动从 Hugging Face 的模型库中下载并加载相应的分词器配置和词汇表。
from transformers import AutoTokenizercheckpoint distilbert-base-uncased-finetuned-sst-2-english
tokenizer AutoTokenizer.from_pretrained(checkpoint)raw_inputs [Ive been waiting for a HuggingFace course my whole life.,I hate this so much!,
]
inputs tokenizer(raw_inputs, paddingTrue, truncationTrue, return_tensorspt)
# raw_inputs要进行分词处理的文本列表。
# paddingTrue表示对输入文本进行填充操作使所有输入序列的长度一致。填充操作通常使用特殊的填充标记如 [PAD]来实现以确保所有输入序列可以在同一个批次中进行处理。
# truncationTrue表示对输入文本进行截断操作。如果输入序列的长度超过了模型所能处理的最大长度将对其进行截断只保留前一部分。
# return_tensorspt表示返回 PyTorch 张量格式的输出。这样处理后的输入数据可以直接用于 PyTorch 模型的输入。
print(inputs){input_ids: tensor([[ 101, 1045, 1005, 2310, 2042, 3403, 2005, 1037, 17662, 12172, 2607, 2026, 2878, 2166, 1012, 102],[ 101, 1045, 5223, 2023, 2061, 2172, 999, 102, 0, 0,0, 0, 0, 0, 0, 0]]), attention_mask: tensor([[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],[1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0]])
}输出中包含两个键 input_ids 和 attention_mask其中 input_ids 对应分词之后的 tokens 映射到的数字编号列表而 attention_mask 则是用来标记哪些 tokens 是被填充的这里“1”表示是原文“0”表示是填充字符。
2将预处理好的输入送入模型 所有存储在 HuggingFace Model Hub 上的模型都可以通过 AutoModel / 特定任务类 / Model.from_pretrained() 来加载权重参数可以是 checkpoint 的名称也可以是本地路径预先下载的模型目录。
部分模型的 Hub 页面中可能会包含很多文件以防万一需要全都下载。
预训练模型的本体只包含基础的 Transformer 模块对于给定的输入它会输出一些神经元的值称为 hidden states 或者特征 (features)。对于 NLP 模型来说可以理解为是文本的高维语义表示。这些 hidden states 通常会被输入到其他的模型部分称为 head以完成特定的任务例如送入到分类头中完成文本分类任务。 Transformers 包提供了一个 AutoModel 类和对应的 from_pretrained() 函数。
from transformers import AutoModelcheckpoint distilbert-base-uncased-finetuned-sst-2-english
model AutoModel.from_pretrained(checkpoint)但是对于有些任务我们不会使用 AutoModel 类而是使用针对特定任务的类。例如情感分析任务使用的是AutoModelForSequenceClassification。
from transformers import AutoTokenizer
from transformers import AutoModelForSequenceClassificationcheckpoint distilbert-base-uncased-finetuned-sst-2-english
tokenizer AutoTokenizer.from_pretrained(checkpoint)
model AutoModelForSequenceClassification.from_pretrained(checkpoint)raw_inputs [Ive been waiting for a HuggingFace course my whole life.,I hate this so much!,
]
inputs tokenizer(raw_inputs, paddingTrue, truncationTrue, return_tensorspt)
outputs model(**inputs)
print(outputs.logits.shape)Transformer 模块的的输出是一个维度为 (Batch size, Sequence length, Hidden size) 的三维张量其中 Batch size 表示每次输入的样本文本序列数量即每次输入多少个句子。Sequence length 表示文本序列的长度即每个句子被分为多少个 token。Hidden size 表示每一个 token 经过模型编码后的输出向量语义表示的维度。可以通过属性outputs.last_hidden_state.shape)访问也可以通过键outputs[“last_hidden_state”]甚至索引访问outputs[0]。
另外除了使用 AutoModel 根据 checkpoint 自动加载模型以外我们也可以直接使用模型对应的 Model 类例如 BERT 对应的就是 BertModel
from transformers import BertModelmodel BertModel.from_pretrained(bert-base-cased)(3) 对模型输出进行后处理
由于模型的输出只是一些数值因此并不适合人类阅读。
在深度学习特别是自然语言处理的模型中logits 是一个非常重要的概念。logits 是模型在进行分类或者生成任务时最后一层神经元的原始输出值它是未经过任何归一化处理的数值。在多分类问题中logits 向量的每个元素对应一个类别其数值大小表示模型认为输入属于该类别的相对可能性但这些数值并不是概率值因为它们没有被约束在 0 到 1 之间且所有元素之和也不一定为 1。要将他们转换为概率值还需要让它们经过一个 SoftMax 层。
所有 Transformers 模型都会输出 logits 值因为训练时的损失函数通常会自动结合激活函数例如 SoftMax与实际的损失函数例如交叉熵 cross entropy。
outputs model(**inputs)
print(outputs.logits)tensor([[-1.5607, 1.6123],[ 4.1692, -3.3464]], grad_fnAddmmBackward0)predictions torch.nn.functional.softmax(outputs.logits, dim-1)
print(predictions)tensor([[4.0195e-02, 9.5980e-01],[9.9946e-01, 5.4418e-04]], grad_fnSoftmaxBackward0)最后为了得到对应的标签可以读取模型 config 中提供的 id2label 属性。
print(model.config.id2label)4.2 pipeline
除上述之外我们也可以使用 Transformer 库的 pipeline() 函数它封装了预训练模型和对应的前处理和后处理环节。我们只需输入文本就能得到预期的答案。
Transformers 库将目前的 NLP 任务归纳为几下几类
文本分类例如情感分析、句子对关系判断等对文本中的词语进行分类例如词性标注 (POS)、命名实体识别 (NER) 等文本生成例如填充预设的模板 (prompt)、预测文本中被遮掩掉 (masked) 的词语从文本中抽取答案例如根据给定的问题从一段文本中抽取出对应的答案根据输入文本生成新的句子例如文本翻译、自动摘要等。
目前常用的 pipelines 有
feature-extraction 获得文本的向量化表示fill-mask 填充被遮盖的词、片段ner命名实体识别question-answering 自动问答sentiment-analysis 情感分析summarization 自动摘要text-generation 文本生成translation 机器翻译zero-shot-classification 零训练样本分类
pipeline 会自动选择合适的预训练模型来完成任务。Transformers 库会在创建对象时下载并且缓存模型只有在首次加载模型时才会下载后续会直接调用缓存好的模型。
我们也可以指定要使用的模型。例如对于文本生成任务我们可以在 Model Hub 页面左边选择 Text Generation tag 查询支持的模型。
generator pipeline(text-generation, modeldistilgpt2)4.2.1 情感分析
我们只需要输入文本就可以得到其情感标签积极/消极以及对应的概率。
from transformers import pipelineclassifier pipeline(sentiment-analysis)
result classifier(Ive been waiting for a HuggingFace course my whole life.)
print(result)
results classifier([Ive been waiting for a HuggingFace course my whole life., I hate this so much!]
)
print(results)No model was supplied, defaulted to distilbert-base-uncased-finetuned-sst-2-english (https://huggingface.co/distilbert-base-uncased-finetuned-sst-2-english)[{label: POSITIVE, score: 0.9598048329353333}]
[{label: POSITIVE, score: 0.9598048329353333}, {label: NEGATIVE, score: 0.9994558691978455}]4.2.2 零训练样本分类
允许我们在不提供任何标注数据的情况下自定义分类标签。
from transformers import pipelineclassifier pipeline(zero-shot-classification)
result classifier(
This is a course about the Transformers library,
candidate_labels[education, politics, business],
)
print(result)No model was supplied, defaulted to facebook/bart-large-mnli (https://huggingface.co/facebook/bart-large-mnli){sequence: This is a course about the Transformers library, labels: [education, business, politics], scores: [0.8445973992347717, 0.11197526752948761, 0.043427325785160065]}4.2.3 文本生成
首先根据任务构建一个模板 (prompt)然后将其送入到模型中来生成后续文本。注意由于文本生成具有随机性因此每次运行都会得到不同的结果。这种模板被称为前缀模板 (Prefix Prompt)。
from transformers import pipelinegenerator pipeline(text-generation)
results generator(In this course, we will teach you how to)
print(results)
results generator(In this course, we will teach you how to,num_return_sequences2,max_length50
)
print(results)No model was supplied, defaulted to gpt2 (https://huggingface.co/gpt2)[{generated_text: In this course, we will teach you how to use data and models that can be applied in any real-world, everyday situation. In most cases, the following will work better than other courses Ive offered for an undergrad or student. In order}]
[{generated_text: In this course, we will teach you how to make your own unique game called Mono from scratch by doing a game engine, a framework and the entire process starting with your initial project. We are planning to make some basic gameplay scenarios and}, {generated_text: In this course, we will teach you how to build a modular computer, how to run it on a modern Windows machine, how to install packages, and how to debug and debug systems. We will cover virtualization and virtualization without a programmer,}]4.2.4 遮盖词填充
给定一段部分词语被遮盖掉 (masked) 的文本使用预训练模型来预测能够填充这些位置的词语。也需要先构建模板然后运用模型来完善模板称为填充模板 (Cloze Prompt)。
from transformers import pipelineunmasker pipeline(fill-mask)
results unmasker(This course will teach you all about mask models., top_k2)
print(results)No model was supplied, defaulted to distilroberta-base (https://huggingface.co/distilroberta-base)[{sequence: This course will teach you all about mathematical models., score: 0.19619858264923096, token: 30412, token_str: mathematical}, {sequence: This course will teach you all about computational models., score: 0.04052719101309776, token: 38163, token_str: computational}]4.2.5 命名实体识别
负责从文本中抽取出指定类型的实体例如人物、地点、组织等。
from transformers import pipelinener pipeline(ner, grouped_entitiesTrue)
results ner(My name is Sylvain and I work at Hugging Face in Brooklyn.)
print(results)No model was supplied, defaulted to dbmdz/bert-large-cased-finetuned-conll03-english (https://huggingface.co/dbmdz/bert-large-cased-finetuned-conll03-english)[{entity_group: PER, score: 0.9981694, word: Sylvain, start: 11, end: 18}, {entity_group: ORG, score: 0.97960186, word: Hugging Face, start: 33, end: 45}, {entity_group: LOC, score: 0.99321055, word: Brooklyn, start: 49, end: 57}]模型正确地识别出了 Sylvain 是一个人物Hugging Face 是一个组织Brooklyn 是一个地名。
通过设置参数 grouped_entitiesTrue使得 pipeline 自动合并属于同一个实体的多个子词 (token)例如这里将“Hugging”和“Face”合并为一个组织实体实际上 Sylvain 也进行了子词合并因为分词器会将 Sylvain 切分为 S、##yl 、##va 和 ##in 四个 token。
4.2.6 自动问答
根据给定的上下文回答问题。
from transformers import pipelinequestion_answerer pipeline(question-answering)
answer question_answerer(questionWhere do I work?,contextMy name is Sylvain and I work at Hugging Face in Brooklyn,
)
print(answer)No model was supplied, defaulted to distilbert-base-cased-distilled-squad (https://huggingface.co/distilbert-base-cased-distilled-squad){score: 0.6949771046638489, start: 33, end: 45, answer: Hugging Face}这里的自动问答 pipeline 实际上是一个抽取式问答模型即从给定的上下文中抽取答案而不是生成答案。
根据形式的不同自动问答 (QA) 系统可以分为三种
抽取式 QA (extractive QA)假设答案就包含在文档中因此直接从文档中抽取答案多选 QA (multiple-choice QA)从多个给定的选项中选择答案相当于做阅读理解题无约束 QA (free-form QA)直接生成答案文本并且对答案文本格式没有任何限制。
4.2.7 自动摘要
将长文本压缩成短文本并且还要尽可能保留原文的主要信息。
from transformers import pipelinesummarizer pipeline(summarization)
results summarizer(America has changed dramatically during recent years. Not only has the number of graduates in traditional engineering disciplines such as mechanical, civil, electrical, chemical, and aeronautical engineering declined, but in most of the premier American universities engineering curricula now concentrate on and encourage largely the study of engineering science. As a result, there are declining offerings in engineering subjects dealing with infrastructure, the environment, and related issues, and greater concentration on high technology subjects, largely supporting increasingly complex scientific developments. While the latter is important, it should not be at the expense of more traditional engineering.Rapidly developing economies such as China and India, as well as other industrial countries in Europe and Asia, continue to encourage and advance the teaching of engineering. Both China and India, respectively, graduate six and eight times as many traditional engineers as does the United States. Other industrial countries at minimum maintain their output, while America suffers an increasingly serious decline in the number of engineering graduates and a lack of well-educated engineers.
)
print(results)No model was supplied, defaulted to sshleifer/distilbart-cnn-12-6 (https://huggingface.co/sshleifer/distilbart-cnn-12-6)[{summary_text: America has changed dramatically during recent years . The number of engineering graduates in the U.S. has declined in traditional engineering disciplines such as mechanical, civil, electrical, chemical, and aeronautical engineering . Rapidly developing economies such as China and India, as well as other industrial countries in Europe and Asia, continue to encourage and advance engineering .}]与文本生成类似我们也可以通过 max_length 或 min_length 参数来控制返回摘要的长度。
