使用场景
用于实体链接任务,通过预训练模型快速识别文本中的实体。
在单跳问答任务中,通过模型生成准确的答案。
应用于多跳问答任务,通过模型的推理过程找到问题的答案。
产品特色
支持生成和检索任务的统一处理,降低部署成本。
在生成过程中实现检索,避免了对查询进行两次前向传递计算。
支持实体链接、单跳问答和多跳问答等多种任务。
提供预训练模型下载,方便用户快速开始。
支持从零开始训练模型,提供灵活的配置选项。
提供详细的评估脚本,方便用户评估模型性能。
使用教程
1. 克隆OneGen仓库到本地环境。
2. 创建并激活Python虚拟环境。
3. 安装所需的依赖包。
4. 下载并解压数据集,准备训练或推理。
5. 根据需要下载预训练模型(可选)。
6. 配置模型参数和路径。
7. 运行推理脚本,进行模型预测。
8. 使用评估脚本,评估模型性能。