SetFit Model 在自定义数据集上训练

Question

我有一个 dataframe 有 3 列，我正在尝试使用 SetFit model 执行一些镜头文本分类

Dataframe (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 model 接受两个输入即文本和 label

我想用额外的输入来训练 model，即 2 个输入 (A,B) 来预测 label (C)。

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

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

Answer 1

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

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

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