moreo
/
kernel_authorship
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

again

This commit is contained in:
Alejandro Moreo Fernandez 2020-05-03 23:08:39 +02:00
parent f118b91ed0
commit 0be3e5547e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/model/classifiers.py
View File

@ -46,7 +46,7 @@ class AuthorshipAttributionClassifier(nn.Module):
ideal_kernel = torch.as_tensor(1 * (np.outer(1 + yi, 1 / (yi + 1)) == 1)).to(self.device) ideal_kernel = torch.as_tensor(1 * (np.outer(1 + yi, 1 / (yi + 1)) == 1)).to(self.device)
loss_sav = KernelAlignmentLoss(kernel, ideal_kernel) loss_sav = KernelAlignmentLoss(kernel, ideal_kernel)
loss = loss_attr + loss_sav loss = loss_attr + 0*loss_sav
loss.backward() loss.backward()
optim.step() optim.step()