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

updated comment for PairwiseComparator

Browse Source
This commit is contained in:
Elaina Claus
2025-12-25 14:53:35 +00:00
parent acbccebb2c
commit 23fddbe5b9
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

@@ -14,7 +14,7 @@ class NumberEmbedder(nn.Module):
def forward(self, x): def forward(self, x):
return self.net(x) return self.net(x)
# 3) Comparator head: takes (ea, eb, e) -> logit for "a > b" # MLP Comparator head: takes (ea, eb, e) -> logit for "a > b"
class PairwiseComparator(nn.Module): class PairwiseComparator(nn.Module):
def __init__(self, d=2, hidden=4, k=0.5): def __init__(self, d=2, hidden=4, k=0.5):
super().__init__() super().__init__()