---

许可证: cc-by-sa-4.0

更新通知

模型权重于2024年2月7日UTC早上7点更新。新的模型权重带来了性能上的显著提升——尤其是在连接操作方面。

如果您在此之前下载了模型，请重新下载权重以获得最佳性能。

SQLCoder-7B-2模型卡片

一种自然语言生成SQL的高能力大型语言模型。

模型详细信息

模型描述

这是Hub上已推送到transformers模型的模型卡片。这个模型卡片是自动生成的。

• 由以下公司开发：Defog, Inc
  
• 模型类型：[文本到SQL]
  
• 许可证：[CC-by-SA-4.0]
  
• 微调自模型：[CodeLlama-7B]
  

模型来源 [可选]

• HuggingFace
  
• GitHub
  
• 演示
  

用途

此模型旨在供非技术用户理解SQL数据库中的数据。它旨在作为一种分析工具，而不是数据库管理工具。

此模型尚未针对拒绝具有对数据库写访问权限的用户的不良请求进行训练，应仅供具有只读访问权限的用户使用。

如何开始使用模型

使用此处的代码来开始使用模型。

提示

请使用以下提示以获得最佳结果。请记住使用 do_sample=False 和 num_beams=4 以获得最佳结果。

### Task
Generate a SQL query to answer [QUESTION]{user_question}[/QUESTION]

### Database Schema
The query will run on a database with the following schema:
{table_metadata_string_DDL_statements}

### Answer
Given the database schema, here is the SQL query that [QUESTION]{user_question}[/QUESTION]
[SQL]

评估

此模型在Defog开发的用于测试和验证模型能力的基于PostgreSQL的评估框架SQL-Eval上进行了评估。

您可以在此处了解更多关于SQLEval背后的方法这里。

结果

我们将每个生成的疑问分为6个类别之一。表格显示了每个模型按类别解答正确问题的百分比。

	日期	分组	排序	比率	连接	where
sqlcoder-70b	96	91.4	97.1	85.7	97.1	91.4
sqlcoder-7b-2	96	91.4	94.3	91.4	94.3	77.1
sqlcoder-34b	80	94.3	85.7	77.1	85.7	80
gpt-4	72	94.3	97.1	80	91.4	80
gpt-4-turbo	76	91.4	91.4	62.8	88.6	77.1
natural-sql-7b	56	88.6	85.7	60	88.6	80
sqlcoder-7b	64	82.9	74.3	54.3	74.3	74.3
gpt-3.5	72	77.1	82.8	34.3	65.7	71.4
claude-2	52	71.4	74.3	57.1	65.7	62.9

模型卡片联系方式

在@defogdata的X上联系我们，或在founders@defog.ai的电子邮件上联系我们。