divide by zero 2

2019-12-10 16:48:13 +01:00 · 2019-12-10 16:48:13 +01:00 · 062f343664
parent 2bab6faf21
commit 062f343664
1 changed files with 1 additions and 1 deletions
--- a/src/data/embeddings.py
+++ b/src/data/embeddings.py
@ -194,6 +194,6 @@ def embedding_matrix(path, voc, lang):

 def WCE_matrix(Xtr, Ytr, lang):
    print('\n# [supervised-matrix]')
-    S = get_supervised_embeddings(Xtr[lang], Ytr[lang])
+    S = get_supervised_embeddings(Xtr[lang], Ytr[lang], max_label_space=50)
    print(f'[embedding matrix done] of shape={S.shape}\n')
    return S