cngthnh/Thesis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

test

Browse Source
This commit is contained in:
thanhvc3
2024-06-19 00:12:43 +07:00
parent 7448528eec
commit 6d43b88599
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
main.py
View File

@@ -479,6 +479,7 @@ class Main(object):
sub, rel, obj, nt_rel, label, neg_ent, sub_samp = self.read_batch( sub, rel, obj, nt_rel, label, neg_ent, sub_samp = self.read_batch(
batch, 'train') batch, 'train')
print(nt_rel)
pred = self.model.forward(sub, rel, nt_rel, neg_ent, self.p.train_strategy) pred = self.model.forward(sub, rel, nt_rel, neg_ent, self.p.train_strategy)
loss = self.model.loss(pred, label, sub_samp) loss = self.model.loss(pred, label, sub_samp)