自定义数据集上的 SetFit 模型训练答案

【问题标题】：SetFit Model training on a custom dataset自定义数据集上的 SetFit 模型训练
【发布时间】：2023-01-31 14:02:26
【问题描述】：

我有一个包含 3 列的数据框，我正在尝试使用 SetFit 模型对几个镜头文本进行分类

数据框 (df)

           A                   B                      C
0    Lorem ipsum ta      lorem ipsum                  Yes
1    Excepteur sint      occaecat excepteur           No
2    Duis aute irure     aute irure                   Yes

传统的 SetFit 模型接受两个输入，即文本和标签

我还想用额外的输入来训练模型，即 2 个输入 (A,B) 来预测标签 (C)。

而且我不确定如何应用 Transformers 库中的 SetFit 模型来执行此操作

我正在尝试搜索如何执行此操作并尝试不同的代码片段，但与此同时我将不胜感激任何指导，提前谢谢您！

【问题讨论】：

标签： python nlp huggingface-transformers transformer-model

【解决方案1】：

SetFit 接受两个输入：文本和标签。

您可以连接 A 列和 B 列中的文本并将其作为 text 输入传递，并将 C 列用于 label 输入。

df['text'] = df['A'] + "_" + df['B']

【讨论】：