Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -225,31 +225,31 @@ RED_SVD = "preprocessor/truncated_svd.joblib"
|
|
| 225 |
|
| 226 |
# Map each model to its specific file paths
|
| 227 |
MODEL_CONFIG = {
|
| 228 |
-
"XGBoost": {
|
| 229 |
"model_path": "models/bow_models_without_pca/xgboost_model.joblib",
|
| 230 |
"vec_path": VEC_COUNT,
|
| 231 |
"red_path": None,
|
| 232 |
"dense_required": False
|
| 233 |
},
|
| 234 |
-
"LightGBM": {
|
| 235 |
"model_path": "models/bow_models_without_pca/lightgbm_model.joblib",
|
| 236 |
"vec_path": VEC_COUNT,
|
| 237 |
"red_path": None,
|
| 238 |
"dense_required": False
|
| 239 |
},
|
| 240 |
-
"Random Forest": {
|
| 241 |
"model_path": "models/bow_models_without_pca/random_forest_model.joblib",
|
| 242 |
"vec_path": VEC_COUNT,
|
| 243 |
"red_path": None,
|
| 244 |
"dense_required": False
|
| 245 |
},
|
| 246 |
-
"Linear SVM": {
|
| 247 |
"model_path": "models/tfidf_models_with_truncatedSVD/linear_svm_model.joblib",
|
| 248 |
"vec_path": VEC_TFIDF,
|
| 249 |
"red_path": RED_SVD,
|
| 250 |
"dense_required": False
|
| 251 |
},
|
| 252 |
-
"Logistic Regression": {
|
| 253 |
"model_path": "models/tfidf_models_with_truncatedSVD/logistic_regression_model.joblib",
|
| 254 |
"vec_path": VEC_TFIDF,
|
| 255 |
"red_path": RED_SVD,
|
|
|
|
| 225 |
|
| 226 |
# Map each model to its specific file paths
|
| 227 |
MODEL_CONFIG = {
|
| 228 |
+
"XGBoost (BoW)": {
|
| 229 |
"model_path": "models/bow_models_without_pca/xgboost_model.joblib",
|
| 230 |
"vec_path": VEC_COUNT,
|
| 231 |
"red_path": None,
|
| 232 |
"dense_required": False
|
| 233 |
},
|
| 234 |
+
"LightGBM (BoW)": {
|
| 235 |
"model_path": "models/bow_models_without_pca/lightgbm_model.joblib",
|
| 236 |
"vec_path": VEC_COUNT,
|
| 237 |
"red_path": None,
|
| 238 |
"dense_required": False
|
| 239 |
},
|
| 240 |
+
"Random Forest (BoW)": {
|
| 241 |
"model_path": "models/bow_models_without_pca/random_forest_model.joblib",
|
| 242 |
"vec_path": VEC_COUNT,
|
| 243 |
"red_path": None,
|
| 244 |
"dense_required": False
|
| 245 |
},
|
| 246 |
+
"Linear SVM (TF-IDF + SVD)": {
|
| 247 |
"model_path": "models/tfidf_models_with_truncatedSVD/linear_svm_model.joblib",
|
| 248 |
"vec_path": VEC_TFIDF,
|
| 249 |
"red_path": RED_SVD,
|
| 250 |
"dense_required": False
|
| 251 |
},
|
| 252 |
+
"Logistic Regression (TF-IDF + SVD)": {
|
| 253 |
"model_path": "models/tfidf_models_with_truncatedSVD/logistic_regression_model.joblib",
|
| 254 |
"vec_path": VEC_TFIDF,
|
| 255 |
"red_path": RED_SVD,
|