diff --git a/app.py b/app.py index 5e31f5d..4040dd9 100644 --- a/app.py +++ b/app.py @@ -7,7 +7,7 @@ clf = joblib.load('./models/logistic_regression.pkl') #comentario = "Me quiero suicidar este día no puede ser peor que el de ayer mi padre me odio y mi madre no me quiere ver" -comentario = "Me quiero suicidar este día no puede ser peor que el de ayer mi padre me odio y mi madre no me quiere ver" +comentario = "Hoy juega Chelsea vs Barcelona y me siento emocionado" # Preprocesado del comentario comentario_procesado = pp.preprocesamiento_con_ortografia(comentario) if comentario_procesado == "": diff --git a/depresion.model b/depresion.model index f3f4bd9..ed59642 100644 Binary files a/depresion.model and b/depresion.model differ diff --git a/model_word2vec_service.py b/model_word2vec_service.py index b63259a..9e8838a 100644 --- a/model_word2vec_service.py +++ b/model_word2vec_service.py @@ -61,15 +61,15 @@ def get_word_vectors(self, corpus): :return: La palabra vector para la palabra. """ array_result = [] - # TODO en error añadir al vocabulario y volver a llamar a la función for word in corpus: try: array_result.append(self.model.wv[word]) - except: - array_result.append(self.getVector250()) + except Exception as e: + return self.getVector250() return array_result def get_cosine_similarity_BECK(self, corpus): + self.add_corpus(corpus) beck = self.get_beck() data = [] for item in beck.keys():