许可协议: cc-by-sa-4.0 库名称: transformers 任务标签: 文本生成

更新通知

模型权重已于2024年2月7日UTC时间上午7点更新。新版模型权重显著提升了模型性能——尤其在联表查询方面表现突出。

若您在此时间前下载过该模型，请重新下载权重以获得最佳性能。

SQLCoder-7B-2模型卡

一款强大的自然语言转SQL生成大语言模型。

模型详情

模型描述

这是托管在HuggingFace Hub上的transformers模型卡。本模型卡为自动生成。

• 开发团队: Defog公司
• 模型类型: [文本转SQL]
• 许可协议: [CC-by-SA-4.0]
• 微调基础模型: [CodeLlama-7B]

模型来源[可选]

• HuggingFace:
• GitHub:
• 演示平台:

用途

本模型旨在帮助非技术人员理解其SQL数据库中的数据，主要作为分析工具使用，而非数据库管理工具。

该模型未经过训练以拒绝具有数据库写入权限用户的恶意请求，建议仅限具有只读权限的用户使用。

快速开始

使用此处提供的代码即可快速启动模型。

提示模板

为获得最佳效果，请使用以下提示模板。建议设置do_sample=False和num_beams=4参数。

### 任务
生成SQL查询以回答[问题]{用户问题}[/问题]

### 数据库结构
查询将在包含以下结构的数据库上执行：
{表元数据字符串_DDL语句}

### 答案
根据数据库结构，以下是回答[问题]{用户问题}[/问题]的SQL查询：
[SQL]

评估

本模型采用Defog开发的SQL-Eval评估框架进行测试，该框架基于PostgreSQL，用于检验模型能力与对齐效果。

关于SQLEval方法论的更多细节请参阅此处。

评估结果

我们将每个生成问题归类到6个类别中。下表展示了各模型在各类别中的正确回答率。

联系信息

欢迎通过X平台@defogdata或邮箱founders@defog.ai联系我们。