dusens / llama-3-8B-Instruct-text2sql

9 stars 1 forks source link

项目名称:llama-3-8B-Instruct-text2sql

项目简介

该模型可以接受中文自然语言指令并转换为SQL查询语句。

新闻

[2024/07/18] 发布 llama-3-8B-sqlcorder-chinese-Instruct-gguf ,对中文对话能力进行微调,进行了中文text2sql的微调:hungingface 地址

接下来方向

Model Method Easy Medium Hard Extra All
llama-3-8B-Instruct-text2sql lora 0.938 0.782 0.581 0.524 0.768
qlora 0 0 0 0 0
base 0.297 0.245 0.151 0.095 0.230
base 0 0 0 0 0
Llama2-13B-Chat lora 0.907 0.729 0.552 0.343 0.68
qlora 0.911 0.7 0.552 0.319 0.664
base 0.214 0.177 0.092 0.036 0.149
CodeLlama-13B-Instruct lora 0.94 0.789 0.684 0.404 0.746
qlora 0.94 0.774 0.626 0.392 0.727
base 0.577 0.352 0.201 0.066 0.335
Qwen-14B-Chat lora 0.895 0.702 0.552 0.331 0.663
qlora 0.919 0.744 0.598 0.367 0.701
base 0 0 0 0 0

训练数据

本项目初期使用的训练数据集为 CSPIDER 中文数据集 Spider数据集 BIRD-SQL 数据集,该数据集包含多种数据库环境下的中文到SQL的查询转换样本。 我们计划在后续阶段引入更多样的中文文本到SQL的样本,以增强模型的泛化能力和准确性。

下载地址

模型名称 完整版 LoRA版 GGUF版
llama-3-8B-Instruct-text2sql
(基于Llama-3-Chinese-8B-Instruct-v2)
[[🤗Hugging Face]]()
[🤖ModelScope]
[[wisemodel]]()
[[🤗Hugging Face]]()
[🤖ModelScope]
[[wisemodel]]()
llama-3-8B-sqlcorder-chinese-Instruct-gguf
(基于llama-3-8b-sqlcorder)
[🤗Hugging Face]

贡献者

sensen

许可证

本项目采用 MIT 许可证。详细许可信息可以在项目仓库的LICENSE文件中找到。

如何参与

欢迎对中文处理和SQL生成感兴趣的开发者加入我们的项目。你可以通过 GitHub Issue 或 Pull Request 的方式参与项目贡献。