Views
No views yet
infoncefact (query side trained against fact text)| eval register | R@5 | R@10 | R@50 | R@100 | ndcg@10 | mrr |
|---|---|---|---|---|---|---|
| user | 4.76 | 8.57 | 30.0 | 41.43 | 5.23 | 0.0739 |
| lawer | 13.33 | 25.24 | 47.62 | 64.76 | 14.53 | 0.1633 |
| fact | 24.29 | 33.81 | 62.38 | 75.71 | 22.43 | 0.2757 |
1from sentence_transformers import SentenceTransformer
2
3m = SentenceTransformer("anonymousresearch123/deka-qwen3-emb-0.6b-infonce-fact")
4q = m.encode(["ลูกจ้างถูกเลิกจ้างโดยไม่บอกกล่าวล่วงหน้า"], prompt_name="query")
5d = m.encode(["<คำพิพากษาฎีกา ...>"])
6print(m.similarity(q, d))1import torch, torch.nn.functional as F
2from transformers import AutoModel, AutoTokenizer
3
4tok = AutoTokenizer.from_pretrained("anonymousresearch123/deka-qwen3-emb-0.6b-infonce-fact", padding_side="left")
5net = AutoModel.from_pretrained("anonymousresearch123/deka-qwen3-emb-0.6b-infonce-fact", torch_dtype=torch.bfloat16).eval()
6
7INSTRUCT = "Given a legal case search query, retrieve relevant prior Supreme Court cases"
8texts = [f"Instruct: {INSTRUCT}\nQuery: ลูกจ้างถูกเลิกจ้าง..."] # query side only
9enc = tok(texts, padding=True, truncation=True, max_length=1300, return_tensors="pt")
10with torch.no_grad():
11 h = net(**enc).last_hidden_state
12emb = F.normalize(h[:, -1].float(), p=2, dim=-1)train / eval / test.
training_history.json in this repo holds the full per-epoch log and config.