chap0lin commited on
Commit
8bbe568
·
1 Parent(s): 2037e9d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- try:
125
  aux_vector.append(reloaded_w2v_model.wv[word])
126
- except:
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