five

Nitishsancs/synthetic_text_to_sql-llama2-1k

收藏
Hugging Face2024-06-23 更新2024-06-29 收录
下载链接:
https://hf-mirror.com/datasets/Nitishsancs/synthetic_text_to_sql-llama2-1k
下载链接
链接失效反馈
官方服务:
资源简介:
该数据集包含与SQL查询相关的多个特征,如id、domain、domain_description、sql_complexity、sql_complexity_description、sql_task_type、sql_task_type_description、sql_prompt、sql_context、sql、sql_explanation和text。这些特征用于描述SQL查询的复杂性、任务类型、提示、上下文、SQL语句本身及其解释。数据集包含一个训练集,包含1000个样本,文件大小为2190404字节。

This dataset contains multiple features related to SQL queries, such as id, domain, domain_description, sql_complexity, sql_complexity_description, sql_task_type, sql_task_type_description, sql_prompt, sql_context, sql, sql_explanation, and text. These features are used to describe the complexity of SQL queries, task types, prompts, context, the SQL statements themselves, and their explanations. The dataset includes a training set with 1000 samples and a file size of 2190404 bytes.
提供机构:
Nitishsancs
原始信息汇总

数据集概述

数据集信息

特征

  • id: 数据类型为 int32
  • domain: 数据类型为 string
  • domain_description: 数据类型为 string
  • sql_complexity: 数据类型为 string
  • sql_complexity_description: 数据类型为 string
  • sql_task_type: 数据类型为 string
  • sql_task_type_description: 数据类型为 string
  • sql_prompt: 数据类型为 string
  • sql_context: 数据类型为 string
  • sql: 数据类型为 string
  • sql_explanation: 数据类型为 string
  • text: 数据类型为 string

数据分割

  • train: 包含 1000 个样本,总字节数为 2190404

数据集大小

  • 下载大小: 749343 字节
  • 数据集大小: 2190404 字节

配置

  • config_name: default
    • data_files:
      • split: train
      • path: data/train-*
搜集汇总
数据集介绍
main_image_url
构建方式
在自然语言处理与数据库交互的交叉领域中,Text-to-SQL任务旨在将人类自然语言查询自动转化为可执行的结构化查询语句。该数据集通过合成方式构建,基于LLaMA-2模型生成1,000条高质量的文本-SQL对。每条样本包含唯一的标识符、领域信息及其描述、SQL复杂度及其说明、任务类型及其描述、SQL提示、上下文、SQL语句、解释以及对应的自然语言文本。数据集的构建过程注重多样性与覆盖性,涵盖了多个领域和不同复杂度的SQL查询,从而为模型训练提供丰富的语义与结构特征。
特点
该数据集的核心特点在于其合成数据的高可控性与结构化丰富度。每条样本均包含详细的元数据字段,如domain、sql_complexity和sql_task_type,这些字段辅以描述性说明,使得数据不仅适用于模型训练,还便于进行细粒度的性能分析与错误诊断。1,000条样本的规模虽小,但通过精心设计的合成流程,确保了数据在领域、难度和任务类型上的均衡分布,从而在有限数据量下最大化代表性。此外,每条样本均提供SQL解释,有助于模型学习推理过程,提升可解释性。
使用方法
该数据集可直接用于微调或评估Text-to-SQL模型,尤其适用于基于LLaMA-2等预训练语言模型的场景。使用时,可将train split中的text字段作为模型输入,sql字段作为目标输出,构建序列到序列的训练范式。元数据字段如domain和sql_complexity可用于分层训练或评估,例如按领域或难度级别分析模型表现。数据以Parquet格式存储,可通过HuggingFace Datasets库加载,支持灵活的数据划分与批量处理,便于集成到现有的NLP实验流程中。
背景与挑战
背景概述
在自然语言处理与数据库交叉领域,将非结构化文本精准转化为结构化查询语言(SQL)是智能数据交互的核心挑战。Nitishsancs/synthetic_text_to_sql-llama2-1k数据集由研究者Nitish Sankaran等人基于Llama 2模型合成构建,于2023年发布,旨在为文本到SQL任务提供高质量、多样化的训练与评估样本。该数据集包含1000条精心设计的实例,覆盖多领域、多复杂度与多任务类型的SQL查询,通过引入领域描述、SQL复杂度说明及自然语言解释等元信息,显著提升了模型对语义映射与逻辑结构的理解能力。作为合成数据驱动方法的典型代表,该数据集有效缓解了真实标注数据稀缺的瓶颈,为提升大语言模型在结构化查询生成中的泛化性能奠定了重要基础,并推动了Text-to-SQL领域从规则驱动向数据驱动范式的转型。
当前挑战
当前数据集面临的首要挑战源于合成数据与真实场景间的语义鸿沟,尽管覆盖多领域与多复杂度,但合成样本可能无法完全捕捉真实用户查询中隐含的歧义性与上下文依赖,导致模型在部署时泛化能力不足。其次,数据集规模有限(仅1000条),难以支撑深度神经网络对大规模训练数据的需求,尤其在处理嵌套查询、多表关联等复杂SQL结构时易出现过拟合现象。此外,构建过程中对Llama 2生成质量的高度依赖引入了系统性偏差,例如领域描述与SQL模板的同质性可能限制模型对罕见模式的学习,而人工校验成本高昂则进一步制约了数据集的规模扩展与质量保障,成为制约该领域迈向实用化的关键瓶颈。
常用场景
经典使用场景
在自然语言处理与数据库查询的交叉领域中,Nitishsancs/synthetic_text_to_sql-llama2-1k数据集为文本到SQL转换任务提供了高质量的合成样本。该数据集涵盖多个领域(domain),并细化了SQL复杂度与任务类型描述,使其成为训练和评估大语言模型在结构化查询生成能力的经典基准。研究者通常利用其丰富的sql_prompt与sql_context字段,引导模型理解自然语言问句与数据库模式之间的关系,从而提升跨领域泛化性能。
实际应用
在实际应用中,该数据集可赋能智能客服、商业智能报表自动化及教育辅助工具等场景。例如,企业可基于此训练模型,使非技术用户通过自然语言直接查询销售数据库,降低数据获取门槛。此外,它还能用于开发交互式SQL教学平台,通过模型生成的查询解释(sql_explanation)帮助学习者理解查询逻辑,提升学习效率。
衍生相关工作
该数据集衍生了一系列经典工作,包括基于Llama系列模型的指令微调策略研究、少样本学习下的SQL生成优化,以及跨领域迁移学习框架。研究者利用其结构化标签(如sql_task_type)开展对比实验,探索提示工程与上下文增强对生成准确性的影响。这些工作不仅验证了合成数据在复杂任务中的有效性,还推动了Text-to-SQL领域从规则驱动向数据驱动范式的转变。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务