训练

此文件夹包含各种示例,用于针对特定任务微调 CrossEncoder 模型。

作为入门,我建议您先看一下 MS MARCO 的示例。

有关如何训练您自己的模型的文档,请参阅训练概述

训练示例

  • distillation - 使模型更小、更快、更轻量的示例。

  • ms_marco - 大量在 MS MARCO 信息检索数据集上进行训练的示例脚本。

  • nli - 自然语言推理 (NLI) 数据涉及使用“矛盾”、“蕴含”和“中性”类别进行句子对分类。

  • quora_duplicate_questions - Quora 重复问题是一个大型语料库,包含来自 Quora 社区的重复问题。该文件夹包含如何训练模型以进行重复问题挖掘和语义搜索的示例。

  • rerankers - 在通用信息检索数据集上进行训练的示例脚本。

  • sts - 训练模型最基本的方法是使用语义文本相似度 (STS) 数据。在这里,我们有一个句子对和一个表示语义相似度的分数。