From 062f343664c692e1f377a420db8b3b2afaf91223 Mon Sep 17 00:00:00 2001
From: Alex Moreo <alejandro.moreo@isti.cnr.it>
Date: Tue, 10 Dec 2019 16:48:13 +0100
Subject: [PATCH] divide by zero 2

---
 src/data/embeddings.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/data/embeddings.py b/src/data/embeddings.py
index 0a7aa4c..58a0b64 100644
--- 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