训练
此文件夹包含各种示例,用于针对特定任务微调 CrossEncoder
模型。
作为入门,我建议您先看一下 MS MARCO 的示例。
有关如何训练您自己的模型的文档,请参阅训练概述。
训练示例
distillation - 使模型更小、更快、更轻量的示例。
ms_marco - 大量在 MS MARCO 信息检索数据集上进行训练的示例脚本。
nli - 自然语言推理 (NLI) 数据涉及使用“矛盾”、“蕴含”和“中性”类别进行句子对分类。
quora_duplicate_questions - Quora 重复问题是一个大型语料库,包含来自 Quora 社区的重复问题。该文件夹包含如何训练模型以进行重复问题挖掘和语义搜索的示例。
rerankers - 在通用信息检索数据集上进行训练的示例脚本。
sts - 训练模型最基本的方法是使用语义文本相似度 (STS) 数据。在这里,我们有一个句子对和一个表示语义相似度的分数。