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:16:05 +07:00
parent 6d43b88599
commit a064d12763
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

@@ -102,6 +102,7 @@ class Main(object):
self.p.num_ent = len(self.ent2id) self.p.num_ent = len(self.ent2id)
self.p.num_rel = len(self.rel2id) // 2 self.p.num_rel = len(self.rel2id) // 2
print("Num rel: " + str(self.p.num_rel))
self.p.embed_dim = self.p.k_w * \ self.p.embed_dim = self.p.k_w * \
self.p.k_h if self.p.embed_dim is None else self.p.embed_dim self.p.k_h if self.p.embed_dim is None else self.p.embed_dim