Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -115,15 +115,15 @@ def classify(df, new_column = True):
|
|
115 |
print("Transformado em W2V")
|
116 |
words = list(reloaded_w2v_model.wv.vocab)
|
117 |
item_shape = np.shape(reloaded_w2v_model.wv[words[0]])
|
118 |
-
print(formatted_sentences)
|
119 |
|
120 |
MCTIinput_vector = []
|
121 |
for sentence in formatted_sentences:
|
122 |
aux_vector = []
|
123 |
for word in sentence:
|
124 |
-
|
125 |
aux_vector.append(reloaded_w2v_model.wv[word])
|
126 |
-
|
127 |
aux_vector.append(np.zeros(item_shape))
|
128 |
MCTIinput_vector.append(aux_vector)
|
129 |
del formatted_sentences
|
|
|
115 |
print("Transformado em W2V")
|
116 |
words = list(reloaded_w2v_model.wv.vocab)
|
117 |
item_shape = np.shape(reloaded_w2v_model.wv[words[0]])
|
118 |
+
# print(formatted_sentences)
|
119 |
|
120 |
MCTIinput_vector = []
|
121 |
for sentence in formatted_sentences:
|
122 |
aux_vector = []
|
123 |
for word in sentence:
|
124 |
+
if word in reloaded_w2v_model.wv.vocab:
|
125 |
aux_vector.append(reloaded_w2v_model.wv[word])
|
126 |
+
else:
|
127 |
aux_vector.append(np.zeros(item_shape))
|
128 |
MCTIinput_vector.append(aux_vector)
|
129 |
del formatted_sentences
|