Views
No views yet
TaylorAI/bge-micro-v2를
디컴파일 함수와 소스 함수를 같은 공간에 정렬하도록 contrastive 파인튜닝한 v14
챔피언(epoch-1) 체크포인트입니다.TaylorAI/bge-micro-v2Labradorlabs/bsca-binary-source-aligned-v3-clean-v3 (fingerprint 220d988ac0980dca)bge_v14_fresh_e1_384_vector943346736accac8b)| metric | value |
|---|---|
| source-function MRR | 0.16672 |
| Recall@1 / @10 / @100 / @1000 | 0.133 / 0.250 / 0.342 / 0.392 |
| component MRR | 0.17792 |
| component Recall@1 / @10 / @100 | 0.142 / 0.258 / 0.375 |
v5:both:lighttraining_state.pt(옵티마이저·스케줄러·RNG 상태)가 포함되어 epoch-2 strict resume가 가능합니다.1import torch, torch.nn.functional as F
2from transformers import AutoModel, AutoTokenizer
3
4name = "Labradorlabs/bsca-bge-micro-v2-contrastive-v14-fresh-clean-v3-eb256-t005-e1-384"
5tok = AutoTokenizer.from_pretrained(name)
6model = AutoModel.from_pretrained(name).eval()
7
8def embed(texts):
9 enc = tok(texts, padding=True, truncation=True, max_length=384, return_tensors="pt")
10 with torch.no_grad():
11 out = model(**enc)
12 emb = out.last_hidden_state[:, 0] # CLS
13 return F.normalize(emb, p=2, dim=1)