🐞 fix: error model
This commit is contained in:
parent
f05dd566f6
commit
a0b3cc98b3
|
@ -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
|
Loading…
Reference in New Issue