Transformers 文档

多项选择

Hugging Face's logo
加入 Hugging Face 社区

并获得增强的文档体验

开始使用

多项选择

多项选择任务类似于问答,不同之处在于它提供了几个候选答案以及上下文,并且模型经过训练以选择正确的答案。

本指南将向您展示如何

  1. BERT 上微调 SWAG 数据集的 regular 配置,以在给定多个选项和一些上下文的情况下选择最佳答案。
  2. 使用您微调的模型进行推理。

在开始之前,请确保您已安装所有必要的库

pip install transformers datasets evaluate

我们鼓励您登录您的 Hugging Face 帐户,以便您可以上传模型并与社区分享。出现提示时,输入您的令牌以登录

>>> from huggingface_hub import notebook_login

>>> notebook_login()

加载 SWAG 数据集

首先从 🤗 Datasets 库加载 SWAG 数据集的 regular 配置

>>> from datasets import load_dataset

>>> swag = load_dataset("swag", "regular")

然后看一下示例

>>> swag["train"][0]
{'ending0': 'passes by walking down the street playing their instruments.',
 'ending1': 'has heard approaching them.',
 'ending2': "arrives and they're outside dancing and asleep.",
 'ending3': 'turns the lead singer watches the performance.',
 'fold-ind': '3416',
 'gold-source': 'gold',
 'label': 0,
 'sent1': 'Members of the procession walk down the street holding small horn brass instruments.',
 'sent2': 'A drum line',
 'startphrase': 'Members of the procession walk down the street holding small horn brass instruments. A drum line',
 'video-id': 'anetv_jkn6uvmqwh4'}

虽然看起来这里有很多字段,但实际上非常简单

  • sent1sent2:这些字段显示句子的开头方式,如果将两者放在一起,您将得到 startphrase 字段。
  • ending:建议句子可能如何结尾,但其中只有一个是正确的。
  • label:标识正确的句子结尾。

预处理

下一步是加载 BERT 分词器来处理句子开头和四个可能的结尾

>>> from transformers import AutoTokenizer

>>> tokenizer = AutoTokenizer.from_pretrained("google-bert/bert-base-uncased")

您要创建的预处理函数需要执行以下操作

  1. 制作 sent1 字段的四个副本,并将每个副本与 sent2 组合以重新创建句子的开头方式。
  2. sent2 与四个可能的句子结尾中的每一个组合。
  3. 展平这两个列表以便您可以对其进行分词,然后在之后取消展平,以便每个示例都具有相应的 input_idsattention_masklabels 字段。
>>> ending_names = ["ending0", "ending1", "ending2", "ending3"]


>>> def preprocess_function(examples):
...     first_sentences = [[context] * 4 for context in examples["sent1"]]
...     question_headers = examples["sent2"]
...     second_sentences = [
...         [f"{header} {examples[end][i]}" for end in ending_names] for i, header in enumerate(question_headers)
...     ]

...     first_sentences = sum(first_sentences, [])
...     second_sentences = sum(second_sentences, [])

...     tokenized_examples = tokenizer(first_sentences, second_sentences, truncation=True)
...     return {k: [v[i : i + 4] for i in range(0, len(v), 4)] for k, v in tokenized_examples.items()}

要将预处理函数应用于整个数据集,请使用 🤗 Datasets map 方法。您可以通过设置 batched=True 一次处理数据集的多个元素来加速 map 函数

>>> tokenized_swag = swag.map(preprocess_function, batched=True)

要创建一批示例,更有效的方法是在整理期间动态填充句子到批次中最长的长度,而不是将整个数据集填充到最大长度。DataCollatorForMultipleChoice 展平所有模型输入,应用填充,然后取消展平结果。

>>> from transformers import DataCollatorForMultipleChoice
>>> collator = DataCollatorForMultipleChoice(tokenizer=tokenizer)

评估

在训练期间包含指标通常有助于评估模型的性能。您可以使用 🤗 Evaluate 库快速加载评估方法。对于此任务,加载 accuracy 指标(请参阅 🤗 Evaluate 快速入门 以了解有关如何加载和计算指标的更多信息)

>>> import evaluate

>>> accuracy = evaluate.load("accuracy")

然后创建一个函数,将您的预测和标签传递给 compute 以计算准确率

>>> import numpy as np


>>> def compute_metrics(eval_pred):
...     predictions, labels = eval_pred
...     predictions = np.argmax(predictions, axis=1)
...     return accuracy.compute(predictions=predictions, references=labels)

您的 compute_metrics 函数现在已准备就绪,您将在设置训练时返回它。

训练

Pytorch
隐藏 Pytorch 内容

如果您不熟悉使用 Trainer 微调模型,请查看 此处的 基本教程!

您现在已准备好开始训练您的模型!使用 AutoModelForMultipleChoice 加载 BERT

>>> from transformers import AutoModelForMultipleChoice, TrainingArguments, Trainer

>>> model = AutoModelForMultipleChoice.from_pretrained("google-bert/bert-base-uncased")

此时,仅剩三个步骤

  1. TrainingArguments 中定义您的训练超参数。唯一必需的参数是 output_dir,它指定保存模型的位置。您将通过设置 push_to_hub=True 将此模型推送到 Hub(您需要登录 Hugging Face 才能上传您的模型)。在每个 epoch 结束时,Trainer 将评估准确率并保存训练检查点。
  2. 将训练参数传递给 Trainer 以及模型、数据集、分词器、数据整理器和 compute_metrics 函数。
  3. 调用 train() 以微调您的模型。
>>> training_args = TrainingArguments(
...     output_dir="my_awesome_swag_model",
...     eval_strategy="epoch",
...     save_strategy="epoch",
...     load_best_model_at_end=True,
...     learning_rate=5e-5,
...     per_device_train_batch_size=16,
...     per_device_eval_batch_size=16,
...     num_train_epochs=3,
...     weight_decay=0.01,
...     push_to_hub=True,
... )

>>> trainer = Trainer(
...     model=model,
...     args=training_args,
...     train_dataset=tokenized_swag["train"],
...     eval_dataset=tokenized_swag["validation"],
...     processing_class=tokenizer,
...     data_collator=collator,
...     compute_metrics=compute_metrics,
... )

>>> trainer.train()

训练完成后,使用 push_to_hub() 方法将您的模型分享到 Hub,以便每个人都可以使用您的模型

>>> trainer.push_to_hub()
TensorFlow
隐藏 TensorFlow 内容

如果您不熟悉使用 Keras 微调模型,请查看 此处的 基本教程!

要在 TensorFlow 中微调模型,首先设置优化器函数、学习率计划和一些训练超参数
>>> from transformers import create_optimizer

>>> batch_size = 16
>>> num_train_epochs = 2
>>> total_train_steps = (len(tokenized_swag["train"]) // batch_size) * num_train_epochs
>>> optimizer, schedule = create_optimizer(init_lr=5e-5, num_warmup_steps=0, num_train_steps=total_train_steps)

然后,您可以使用 TFAutoModelForMultipleChoice 加载 BERT

>>> from transformers import TFAutoModelForMultipleChoice

>>> model = TFAutoModelForMultipleChoice.from_pretrained("google-bert/bert-base-uncased")

使用 prepare_tf_dataset() 将您的数据集转换为 tf.data.Dataset 格式

>>> data_collator = DataCollatorForMultipleChoice(tokenizer=tokenizer)
>>> tf_train_set = model.prepare_tf_dataset(
...     tokenized_swag["train"],
...     shuffle=True,
...     batch_size=batch_size,
...     collate_fn=data_collator,
... )

>>> tf_validation_set = model.prepare_tf_dataset(
...     tokenized_swag["validation"],
...     shuffle=False,
...     batch_size=batch_size,
...     collate_fn=data_collator,
... )

使用 compile 配置模型以进行训练。请注意,Transformers 模型都具有默认的与任务相关的损失函数,因此您无需指定一个,除非您想要指定一个

>>> model.compile(optimizer=optimizer)  # No loss argument!

在开始训练之前要设置的最后两件事是从预测中计算准确率,并提供一种将您的模型推送到 Hub 的方法。这两者都是通过使用 Keras 回调 完成的。

将您的 compute_metrics 函数传递给 KerasMetricCallback

>>> from transformers.keras_callbacks import KerasMetricCallback

>>> metric_callback = KerasMetricCallback(metric_fn=compute_metrics, eval_dataset=tf_validation_set)

PushToHubCallback 中指定推送您的模型和分词器的位置

>>> from transformers.keras_callbacks import PushToHubCallback

>>> push_to_hub_callback = PushToHubCallback(
...     output_dir="my_awesome_model",
...     tokenizer=tokenizer,
... )

然后将您的回调捆绑在一起

>>> callbacks = [metric_callback, push_to_hub_callback]

最后,您已准备好开始训练您的模型!使用您的训练和验证数据集、epoch 数和回调调用 fit 以微调模型

>>> model.fit(x=tf_train_set, validation_data=tf_validation_set, epochs=2, callbacks=callbacks)

训练完成后,您的模型将自动上传到 Hub,以便每个人都可以使用它!

有关如何为多项选择微调模型的更深入示例,请查看相应的 PyTorch notebookTensorFlow notebook

推理

太棒了,现在您已经微调了模型,您可以将其用于推理了!

想出一些文本和两个候选答案

>>> prompt = "France has a bread law, Le Décret Pain, with strict rules on what is allowed in a traditional baguette."
>>> candidate1 = "The law does not apply to croissants and brioche."
>>> candidate2 = "The law applies to baguettes."
Pytorch
隐藏 Pytorch 内容

对每个提示和候选答案对进行分词,并返回 PyTorch 张量。您还应该创建一些 labels

>>> from transformers import AutoTokenizer

>>> tokenizer = AutoTokenizer.from_pretrained("username/my_awesome_swag_model")
>>> inputs = tokenizer([[prompt, candidate1], [prompt, candidate2]], return_tensors="pt", padding=True)
>>> labels = torch.tensor(0).unsqueeze(0)

将您的输入和标签传递给模型并返回 logits

>>> from transformers import AutoModelForMultipleChoice

>>> model = AutoModelForMultipleChoice.from_pretrained("username/my_awesome_swag_model")
>>> outputs = model(**{k: v.unsqueeze(0) for k, v in inputs.items()}, labels=labels)
>>> logits = outputs.logits

获取概率最高的类别

>>> predicted_class = logits.argmax().item()
>>> predicted_class
0
TensorFlow
隐藏 TensorFlow 内容

对每个提示和候选答案对进行分词,并返回 TensorFlow 张量

>>> from transformers import AutoTokenizer

>>> tokenizer = AutoTokenizer.from_pretrained("username/my_awesome_swag_model")
>>> inputs = tokenizer([[prompt, candidate1], [prompt, candidate2]], return_tensors="tf", padding=True)

将您的输入传递给模型并返回 logits

>>> from transformers import TFAutoModelForMultipleChoice

>>> model = TFAutoModelForMultipleChoice.from_pretrained("username/my_awesome_swag_model")
>>> inputs = {k: tf.expand_dims(v, 0) for k, v in inputs.items()}
>>> outputs = model(inputs)
>>> logits = outputs.logits

获取概率最高的类别

>>> predicted_class = int(tf.math.argmax(logits, axis=-1)[0])
>>> predicted_class
0
< > 在 GitHub 上更新