T5 Base Text To Sql

🚀 T5-SQL-Translator

T5-SQL-Translator是Google T5-small模型的微调版本，专注于将英语自然语言查询转换为SQL SELECT查询。该模型经过训练，能够理解英语查询并为数据库生成相应的SQL SELECT查询，对于将自然语言自动转换为SQL（尤其是SELECT操作）具有重要价值。

🚀 快速开始

安装依赖库

pip install transformers

推理示例

# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

tokenizer = AutoTokenizer.from_pretrained("suriya7/t5-base-text-to-sql")
model = AutoModelForSeq2SeqLM.from_pretrained("suriya7/t5-base-text-to-sql")

def translate_to_sql_select(english_query):
    input_text = "translate English to SQL: "english_query
    input_ids = tokenizer.encode(input_text, return_tensors="pt",max_new_tokens=100,do_sample=False)
    outputs = model.generate(input_ids)
    sql_query = tokenizer.decode(outputs[0], skip_special_tokens=True)
    return sql_query

# Example usage
english_query = "Show all employees with salary greater than $50000"
sql_query = translate_to_sql_select(english_query)
print("SQL Query:", sql_query)

✨ 主要特性

• 能够将英语自然语言查询自动转换为SQL SELECT查询，用于数据库查询。

📚 详细文档

模型详情

性能指标

• 评估指标：BLEU分数为34.962700

致谢

• 原始T5模型由Google Research开发。
• 训练数据来源于https://huggingface.co/datasets/b-mc2/sql-create-context
• 特别感谢Hugging Face提供的Transformers库和模型中心，方便模型的共享。

📄 许可证

本项目采用MIT许可证。

📞 联系我们

如果您对该模型有任何疑问或问题，请随时联系：thesuriya3@gmail.com