wassname/Castor
1
0
Fork 0
mirror of https://github.com/wassname/Castor.git synced 2026-09-09 11:13:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix VDPWI focus cube padding (#162)

Browse Source
This commit is contained in:
Ralph Tang
2018-11-12 19:35:11 -05:00
committed by GitHub
parent 951df4ada5
commit f510fa6bae
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
vdpwi/model.py
+6 -6
View File
@@ -77,14 +77,14 @@ class VDPWIModel(nn.Module):
def create_pad_cube(self, sent1, sent2):
pad_cube = []
max_len1 = max([len(s.split()) for s in sent1])
max_len2 = max([len(s.split()) for s in sent2])
sent1_lengths = [len(s.split()) for s in sent1]
sent2_lengths = [len(s.split()) for s in sent2]
max_len1 = max(sent1_lengths)
max_len2 = max(sent2_lengths)
for s1, s2 in zip(sent1, sent2):
pad1 = (max_len1 - len(s1.split()))
pad2 = (max_len2 - len(s2.split()))
for s1_length, s2_length in zip(sent1_lengths, sent2_lengths):
pad_mask = np.ones((max_len1, max_len2))
pad_mask[:len(s1), :len(s2)] = 0
pad_mask[:s1_length, :s2_length] = 0
pad_cube.append(pad_mask)
pad_cube = np.array(pad_cube)