🐞 fix: error model

This commit is contained in:
ja'far shodiq 2025-04-11 18:11:28 +08:00
parent f05dd566f6
commit a0b3cc98b3
1 changed files with 23 additions and 23 deletions

View File

@ -109,28 +109,28 @@ def generate_wordclouds(wordcloud_data: pd.DataFrame, label_colors: dict) -> dic
# Memuat Model dan Prediksi Sentimen # Memuat Model dan Prediksi Sentimen
# ====================================== # ======================================
vectorizer = joblib.load('models/datasets-tfidf.pkl') # vectorizer = joblib.load('models/datasets-tfidf.pkl')
def load_model_and_vectorizer(model_path, vectorizer_path): # def load_model_and_vectorizer(model_path, vectorizer_path):
""" # """
Memuat model dari file pickle. # Memuat model dari file pickle.
""" # """
try: # try:
model = joblib.load(model_path) # model = joblib.load(model_path)
text_vectorizer = joblib.load(vectorizer_path) # text_vectorizer = joblib.load(vectorizer_path)
return model, text_vectorizer # return model, text_vectorizer
except Exception as e: # except Exception as e:
print(f"Error loading model or vectorizer: {e}") # print(f"Error loading model or vectorizer: {e}")
return None, None # return None, None
def predict_sentiment(model, text_vectorizer, text): # def predict_sentiment(model, text_vectorizer, text):
""" # """
Melakukan prediksi sentimen terhadap teks yang diberikan menggunakan model yang dipilih. # Melakukan prediksi sentimen terhadap teks yang diberikan menggunakan model yang dipilih.
""" # """
try: # try:
text_vectorized = text_vectorizer.transform([text]) # text_vectorized = text_vectorizer.transform([text])
prediction = model.predict(text_vectorized) # prediction = model.predict(text_vectorized)
return prediction[0] # return prediction[0]
except Exception as e: # except Exception as e:
print(f"Error predicting sentiment: {e}") # print(f"Error predicting sentiment: {e}")
return None # return None