chore: added training framework constant

aayushacharya · Jan 25, 2024 · 6fcac41 · 6fcac41
1 parent 8ad7d9f
commit 6fcac41
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/evadb/configuration/constants.py b/evadb/configuration/constants.py
@@ -38,3 +38,4 @@
 DEFAULT_XGBOOST_TASK = "regression"
 DEFAULT_SKLEARN_TRAIN_MODEL = "rf"
 SKLEARN_SUPPORTED_MODELS = ["rf", "extra_tree", "kneighbor"]
+TRAINING_FRAMEWORKS = ["Sklearn", "Ludwig", "XGBoost", "Forecasting"]