Spaces:

nishu08
/

sql-error-classifier-train

Sleeping

App Files Files Community

sql-error-classifier-train / README.md

nishu08

Deploy CodeBERT training Space

8464aea verified 14 days ago

preview code

Raw

History Blame Contribute Delete

1.27 kB

metadata

title: SQL Error Classifier Training
emoji: 🧠
colorFrom: blue
colorTo: green
sdk: docker
app_port: 7860
pinned: false
license: mit
hardware: t4-small

SQL Error Classifier — CodeBERT Training Space

Train microsoft/codebert-base as a cross-encoder for multi-label SQL error classification.

Setup

Hardware: Settings → Hardware → GPU t4-small (recommended)
Secrets: Settings → Secrets → add HF_TOKEN (Hugging Face write token) to push models to your account
Data: Include data/sql_errors_dev.parquet in this Space repo, or upload parquet at runtime

Usage

Choose bundled dataset or upload your own parquet
Set epochs, batch size, max samples
Click Start Training
Optionally enable Push to Hub with model id your-username/sql-codebert-classifier

Dataset columns

Required (aliases supported):

Column	Aliases
`question`	—
`schema`	—
`student_sql`	`query`
`correct_sql`	`correct_query`
`error_labels`	`label_name`

Labels (9-class multi-label)

JOIN_ERROR, AGGREGATION_ERROR, FILTER_ERROR, WINDOW_FUNCTION_ERROR, SUBQUERY_ERROR, NULL_HANDLING_ERROR, PERFORMANCE_ERROR, LOGICAL_ERROR, SYNTAX_ERROR