Views
No views yet
| Metric | Result | Target |
|---|---|---|
| MRR | 0.9052 | 0.60 |
| Hits@1 | 86.2% | - |
| Hits@5 | 95.9% | - |
| Hits@10 | 97.3% | - |
| Median Rank | 1.0 | - |
1from transformers import AutoModel, AutoTokenizer
2
3# 1. Load Model
4model = AutoModel.from_pretrained("uddeshya-k/RepoJepa", trust_remote_code=True)
5tokenizer = AutoTokenizer.from_pretrained("microsoft/codebert-base")
6
7
8# 2. Encode Code
9code = "def handle_login(user): return auth.verify(user)"
10code_embed = model.encode_code(**tokenizer(code, return_tensors="pt"))
11
12# 3. Encode Query
13query = "how to authenticate users?"
14query_embed = model.encode_query(**tokenizer(query, return_tensors="pt"))
15
16# 4. Search
17similarity = (code_embed @ query_embed.T).item()
18print(f"Similarity: {similarity:.4f}")