No commit message

muhaochen · Nov 16, 2017 · 245563d · 245563d
1 parent 46d6685
commit 245563d
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/run/en_de/test_MMTransE_lan_mapping_120k_fk.py b/run/en_de/test_MMTransE_lan_mapping_120k_fk.py
@@ -88,7 +88,7 @@ def test(model, vocab, index, src_lan, tgt_lan, map, score, past_num):
         for i in range(len(cand)):
             last_i = i
             tmp_cand = cand[i]
-            if hit == False and (seem_hit(tmp_cand, tgt) == True or tmp_cand[0] == word):
+            if hit == False and (seem_hit(tmp_cand, tgt) == True or tmp_cand == word):
                 hit = True
             if hit == True:
                 tmp_score[i] = 1.0

diff --git a/run/en_de/test_MMTransE_lan_mapping_15k_fk.py b/run/en_de/test_MMTransE_lan_mapping_15k_fk.py
@@ -89,7 +89,7 @@ def test(model, vocab, index, src_lan, tgt_lan, map, score, past_num):
         for i in range(len(cand)):
             last_i = i
             tmp_cand = cand[i]
-            if hit == False and (seem_hit(tmp_cand, tgt) == True or tmp_cand[0] == word):
+            if hit == False and (seem_hit(tmp_cand, tgt) == True or tmp_cand == word):
                 hit = True
             if hit == True:
                 tmp_score[i] = 1.0

diff --git a/run/en_de/test_MMTransE_lan_mapping_cn.py b/run/en_de/test_MMTransE_lan_mapping_cn.py
@@ -88,7 +88,7 @@ def test(model, vocab, index, src_lan, tgt_lan, map, score, past_num):
         for i in range(len(cand)):
             last_i = i
             tmp_cand = cand[i]
-            if hit == False and (seem_hit(tmp_cand, tgt) == True or tmp_cand[0] == word):
+            if hit == False and (seem_hit(tmp_cand, tgt) == True or tmp_cand == word):
                 hit = True
             if hit == True:
                 tmp_score[i] = 1.0

diff --git a/run/en_fr/test_MMTransE_lan_mapping_120k_fk.py b/run/en_fr/test_MMTransE_lan_mapping_120k_fk.py
@@ -88,7 +88,7 @@ def test(model, vocab, index, src_lan, tgt_lan, map, score, past_num):
         for i in range(len(cand)):
             last_i = i
             tmp_cand = cand[i]
-            if hit == False and (seem_hit(tmp_cand, tgt) == True or tmp_cand[0] == word):
+            if hit == False and (seem_hit(tmp_cand, tgt) == True or tmp_cand == word):
                 hit = True
             if hit == True:
                 tmp_score[i] = 1.0

diff --git a/run/en_fr/test_MMTransE_lan_mapping_15k_fk.py b/run/en_fr/test_MMTransE_lan_mapping_15k_fk.py
@@ -88,7 +88,7 @@ def test(model, vocab, index, src_lan, tgt_lan, map, score, past_num):
         for i in range(len(cand)):
             last_i = i
             tmp_cand = cand[i]
-            if hit == False and (seem_hit(tmp_cand, tgt) == True or tmp_cand[0] == word):
+            if hit == False and (seem_hit(tmp_cand, tgt) == True or tmp_cand == word):
                 hit = True
             if hit == True:
                 tmp_score[i] = 1.0

diff --git a/run/en_fr/test_MMTransE_lan_mapping_cn.py b/run/en_fr/test_MMTransE_lan_mapping_cn.py
@@ -88,7 +88,7 @@ def test(model, vocab, index, src_lan, tgt_lan, map, score, past_num):
         for i in range(len(cand)):
             last_i = i
             tmp_cand = cand[i]
-            if hit == False and (seem_hit(tmp_cand, tgt) == True or tmp_cand[0] == word):
+            if hit == False and (seem_hit(tmp_cand, tgt) == True or tmp_cand == word):
                 hit = True
             if hit == True:
                 tmp_score[i] = 1.0