From a85cf7d1d980b5db0f03c5887b478b0ad6d6ab3e Mon Sep 17 00:00:00 2001 From: Jonas Gabriel Date: Tue, 5 Mar 2024 19:44:35 -0300 Subject: [PATCH] resolving importing issues --- api/DataProcesser.py | 5 +++-- api/NbEmotionsModel.py | 2 +- api/Neural_Network2.py | 8 ++++---- 3 files changed, 8 insertions(+), 7 deletions(-) diff --git a/api/DataProcesser.py b/api/DataProcesser.py index b0e42f73..06353468 100644 --- a/api/DataProcesser.py +++ b/api/DataProcesser.py @@ -2,7 +2,7 @@ from NbEmotionsModel import make_prediction from NbLinRegressionModel import make_prediction_nblin from available_classifiers import get_available_classifiers -from tensorflow.keras.models import load_model +from tensorflow.python.keras.models import load_model import pandas as pd import numpy as np @@ -11,8 +11,9 @@ import re import joblib import string -from nltk.corpus import stopwords + import nltk +from nltk.corpus import stopwords # mais imports class DataProcesser(): diff --git a/api/NbEmotionsModel.py b/api/NbEmotionsModel.py index 4bf58cc2..49b970a5 100644 --- a/api/NbEmotionsModel.py +++ b/api/NbEmotionsModel.py @@ -14,7 +14,7 @@ def make_prediction(my_sentence): pipeline = pickle.load(model_file) # Fazendo previsões para os textos - predictions = pipeline.predict([texts]) + predictions = pipeline.predict([my_sentence]) return predictions[0] diff --git a/api/Neural_Network2.py b/api/Neural_Network2.py index a7e64705..4c06fab4 100644 --- a/api/Neural_Network2.py +++ b/api/Neural_Network2.py @@ -6,11 +6,11 @@ import tensorflow as tf import joblib -from tensorflow.keras import layers -from tensorflow.keras.layers import TextVectorization -from tensorflow.keras.models import Sequential +from tensorflow.python.keras import layers +from tensorflow.python.keras.layers import TextVectorization +from tensorflow.python.keras.models import Sequential from sklearn.preprocessing import LabelEncoder -from tensorflow.keras.callbacks import Callback +from tensorflow.python.keras.callbacks import Callback def preprocess_text(text): text = text.lower()