Views
No views yet
cross-encoder/ms-marco-MiniLM-L-12-v2| Strategy | Params | NDCG@10 | Delta |
|---|---|---|---|
| A: Circuit MLP-only | 2.36M | 0.6545 | +0.0110 |
| B: Last-4 Layers | 7.10M | 0.6686 | +0.0251 |
| C: Full Fine-Tuning | 33.36M | 0.6879 | +0.0444 |
| D: Circuit-Full (BM25) | 4.73M | 0.6707 | +0.0272 |
| E: Circuit-Full (Mixed) | 4.73M | 0.6622 | +0.0187 |
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4tokenizer = AutoTokenizer.from_pretrained("JAYADIR/mdts-circuit-full-bm25")
5model = AutoModelForSequenceClassification.from_pretrained("JAYADIR/mdts-circuit-full-bm25")
6
7query = "What fertilizer is best for wheat?"
8passage = "Wheat requires nitrogen-rich fertilizer during early growth stages."
9
10inputs = tokenizer(query, passage, return_tensors="pt", truncation=True, max_length=512)
11with torch.no_grad():
12 score = model(**inputs).logits.squeeze().item()
13print(f"Relevance score: {score:.4f}")