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