naufalnashif
commited on
Commit
·
0c9cbe9
1
Parent(s):
7b3f7eb
Update app.py
Browse files
app.py
CHANGED
@@ -85,14 +85,14 @@ tfidf_model_path = 'X_tfidf_model.joblib'
|
|
85 |
tfidf_vectorizer = joblib.load(tfidf_model_path)
|
86 |
|
87 |
# Tambahkan widget untuk memilih model
|
88 |
-
selected_model = st.selectbox("Pilih Model Sentimen:", ("
|
89 |
|
90 |
# Fungsi untuk memilih model berdasarkan pilihan pengguna
|
91 |
def select_sentiment_model(selected_model):
|
92 |
-
if selected_model == "
|
93 |
-
model_path = 'best_rf_model_smote.joblib'
|
94 |
-
elif selected_model == "Ensemble":
|
95 |
model_path = 'ensemble_clf_soft_smote.joblib'
|
|
|
|
|
96 |
elif selected_model == "Naive Bayes":
|
97 |
model_path = 'naive_bayes_model_smote.joblib'
|
98 |
elif selected_model == "Logistic Regression":
|
|
|
85 |
tfidf_vectorizer = joblib.load(tfidf_model_path)
|
86 |
|
87 |
# Tambahkan widget untuk memilih model
|
88 |
+
selected_model = st.selectbox("Pilih Model Sentimen:", ("Ensemble", "Naive Bayes", "Logistic Regression"))
|
89 |
|
90 |
# Fungsi untuk memilih model berdasarkan pilihan pengguna
|
91 |
def select_sentiment_model(selected_model):
|
92 |
+
if selected_model == "Ensemble":
|
|
|
|
|
93 |
model_path = 'ensemble_clf_soft_smote.joblib'
|
94 |
+
elif selected_model == "Random Forest":
|
95 |
+
model_path = 'best_rf_model_smote.joblib'
|
96 |
elif selected_model == "Naive Bayes":
|
97 |
model_path = 'naive_bayes_model_smote.joblib'
|
98 |
elif selected_model == "Logistic Regression":
|