Fixed the problem of index out of range when n_src<i<max_rank

chenyahao4179 · Oct 29, 2018 · 7223562 · 7223562
1 parent 1312aca
commit 7223562
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/dico_builder.py b/src/dico_builder.py
@@ -26,7 +26,7 @@ def get_candidates(emb1, emb2, params):
     # number of source words to consider
     n_src = emb1.size(0)
     if params.dico_max_rank > 0 and not params.dico_method.startswith('invsm_beta_'):
-        n_src = params.dico_max_rank
+        n_src = min(params.dico_max_rank, n_src)
 
     # nearest neighbors
     if params.dico_method == 'nn':