naufalnashif commited on
Commit
0c9cbe9
·
1 Parent(s): 7b3f7eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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:", ("Random Forest", "Naive Bayes", "Logistic Regression", "Ensemble"))
89
 
90
  # Fungsi untuk memilih model berdasarkan pilihan pengguna
91
  def select_sentiment_model(selected_model):
92
- if selected_model == "Random Forest":
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":