nivirx/mltoys
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

default k=0.5

Browse Source
This commit is contained in:
Elaina Claus
2025-12-23 10:14:08 -05:00
parent 6e31865a84
commit 9ea8ef3458
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pairwise_comp_nn.py
View File

@@ -16,7 +16,7 @@ class NumberEmbedder(nn.Module):
# 3) Comparator head: takes (ea, eb, e) -> logit for "a > b" # 3) Comparator head: takes (ea, eb, e) -> logit for "a > b"
class PairwiseComparator(nn.Module): class PairwiseComparator(nn.Module):
def __init__(self, d=4, hidden=16, k=1.0): def __init__(self, d=4, hidden=16, k=0.5):
super().__init__() super().__init__()
self.log_k = nn.Parameter(torch.tensor([k])) self.log_k = nn.Parameter(torch.tensor([k]))
self.embed = NumberEmbedder(d, hidden) self.embed = NumberEmbedder(d, hidden)