migration-copilot-qwen2-5-coder-1-5b-instruct

Fine-tuned Qwen2.5-Coder-1.5B-Instruct for enterprise SQL/HiveQL/PL-SQL/Stored Procedure โ†’ PySpark migration.

Part of the Enterprise Migration Copilot project.

Benchmark Results

Evaluated on 480 held-out scripts (120 per language), never seen during training:

Language Pass Rate
SQL 48%
HiveQL 61%
PL/SQL 50%
Stored Procedure 22%
Overall 45%

Metrics: syntax_valid AND has_pyspark_ops AND semantic_sim (60% table name coverage).

Training Details

  • Base model: Qwen/Qwen2.5-Coder-1.5B-Instruct (1.5B parameters)
  • Method: LoRA fine-tuning (r=16, alpha=32)
  • Training data: 1,312 validated SQLโ†’PySpark pairs
  • Data sources: 300 hand-crafted Claude examples (99.67% validation pass rate) + 1,012 Ollama-generated pairs (79.58% pass rate)
  • Languages: SQL, HiveQL, PL/SQL, Stored Procedures (T-SQL)
  • Epochs: 3
  • Train loss: 0.387 โ†’ 0.307
  • Eval loss: 0.390 โ†’ 0.344
  • Hardware: Google Colab T4 GPU (free tier)
  • Training time: ~21 minutes

Prompt Format

### Instruction:
Convert the following {SOURCE_LANGUAGE} code to PySpark.
Difficulty: {difficulty}

### Input:
{source_code}

### Response:

Example

Input (SQL):

SELECT customer_id, SUM(amount) AS total
FROM orders
WHERE status = 'completed'
GROUP BY customer_id
ORDER BY total DESC;

Output (PySpark):

from pyspark.sql import functions as F

df = spark.table('orders')
result = (
    df.filter(F.col('status') == 'completed')
      .groupBy('customer_id')
      .agg(F.sum('amount').alias('total'))
      .orderBy(F.col('total').desc())
)
result.show()

Intended Use

  • Enterprise legacy SQL migration to Apache Spark / Databricks
  • Fintech data pipeline modernization
  • Lightweight deployment (1.5B params, lower memory footprint)

Limitations

  • Lower overall accuracy (45%) vs Phi-3.5-mini (57%) due to smaller base model
  • Stored Procedure migration accuracy is 22% โ€” complex T-SQL constructs require manual review

Links

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Model tree for praveends/migration-copilot-qwen2-5-coder-1-5b-instruct

Finetuned
(184)
this model