Feat:Added model LR

SebastianCB-dev · Dec 4, 2022 · e9a6a15 · e9a6a15
1 parent 1b829c0
commit e9a6a15
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 98 deletions.
diff --git a/app.py b/app.py
@@ -1,8 +1,10 @@
 from preprocessing_service import Preprocesamiento
 from model_word2vec_service import ModelWord2Vec
+import joblib
 
 pp = Preprocesamiento()
 w2v = ModelWord2Vec()
+clf = joblib.load('./models/logistic_regression.pkl')
 
 comentario = "Estoy muy triste y no se que hacer"
 # Escribe un comentario triste y largo
@@ -15,3 +17,5 @@
 # Obtener la respuesta por item basandose en la similitud de coseno
 results_beck = w2v.get_result_beck(cosine_similarity_beck)
 print("El comentario lleno el inventario BECK de esta manera:", results_beck)
+
+print('Predicción:', clf.predict([results_beck]))
diff --git a/models/KNN/test1.py b/models/KNN/test1.py
diff --git a/models/SVM/test1.py b/models/SVM/test1.py
diff --git a/models/logistic_regression.pkl b/models/logistic_regression.pkl