AAP-SQL R1 schema reranker
AAP-SQL R1 是完整 AAP-SQL 設定中的 cross-encoder schema 重排器。它對 E2 取回的候選欄位評分，保留前 10 個欄位供後續提示增強使用。
AAP-SQL R1 is the cross-encoder schema reranker used by the full AAP-SQL configuration. It scores the candidates returned by E2 and retains the top 10 columns for prompt augmentation.
Model details
- Base model:
cross-encoder/ms-marco-MiniLM-L6-v2 - Training objective:
BinaryCrossEntropyLoss - Training seed:
42 - Training data: schema-ranking examples derived from the BIRD training split and schema descriptions
- Expected library:
sentence-transformers>=5.1.2
Use with AAP-SQL
Download this repository into the path expected by the final runner:
hf download TommyPanLab/AAP-SQL-R1 --local-dir models/cross_encoder_schema_paper_repro
Direct loading:
from sentence_transformers import CrossEncoder
model = CrossEncoder("TommyPanLab/AAP-SQL-R1")
scores = model.predict([("user question", "table.column: column description")])
The complete pipeline, required BIRD directory layout, and Gemini 3.1 result are documented in the AAP-SQL publication branch.
Data and license notice
The training examples were derived from the BIRD benchmark. Review the BIRD project terms before using the model. No additional license has been declared for these fine-tuned weights; the upstream model and dataset terms still apply.
- Downloads last month
- 21
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support
Model tree for TommyPanLab/AAP-SQL-R1
Base model
microsoft/MiniLM-L12-H384-uncased Quantized
cross-encoder/ms-marco-MiniLM-L12-v2 Quantized
cross-encoder/ms-marco-MiniLM-L6-v2
from sentence_transformers import CrossEncoder model = CrossEncoder("TommyPanLab/AAP-SQL-R1") query = "Which planet is known as the Red Planet?" passages = [ "Venus is often called Earth's twin because of its similar size and proximity.", "Mars, known for its reddish appearance, is often referred to as the Red Planet.", "Jupiter, the largest planet in our solar system, has a prominent red spot.", "Saturn, famous for its rings, is sometimes mistaken for the Red Planet." ] scores = model.predict([(query, passage) for passage in passages]) print(scores)