From 6fcac41bd2c72f0b94ad38ba55cb9bf901136d9a Mon Sep 17 00:00:00 2001 From: Aayush Acharya Date: Thu, 25 Jan 2024 08:06:45 +0545 Subject: [PATCH] chore: added training framework constant --- evadb/configuration/constants.py | 1 + 1 file changed, 1 insertion(+) diff --git a/evadb/configuration/constants.py b/evadb/configuration/constants.py index 126e6bcfc..c8b84ca93 100644 --- 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"]