andrea.pedrotti
/
gFun
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

divide by zero 2

This commit is contained in:
Alejandro Moreo Fernandez 2019-12-10 16:48:13 +01:00
parent 2bab6faf21
commit 062f343664
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/data/embeddings.py
View File

@ -194,6 +194,6 @@ def embedding_matrix(path, voc, lang):
def WCE_matrix(Xtr, Ytr, lang): def WCE_matrix(Xtr, Ytr, lang):
print('\n# [supervised-matrix]') 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') print(f'[embedding matrix done] of shape={S.shape}\n')
return S return S