Fix: train model with options

traindb-project · May 24, 2024 · 553ff8d · 553ff8d
1 parent 9b49a78
commit 553ff8d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bi-tool/redash/models/train_model.py b/bi-tool/redash/models/train_model.py
@@ -51,11 +51,11 @@ def train_model(cls, **kwargs):
  sql = "TRAIN MODEL "+ kwargs["model_name"] + \
  " MODELTYPE "+ kwargs["model_type"] + \
  " ON "+ target + \
- " OPTIONS ('epochs' = 1)"
- # " OPTIONS ( "+ ", ".join(kwargs["options"]) +" )"
+ " OPTIONS ( "+ kwargs["options"].replace('"', '').replace('{', '').replace('}', '').replace(':', ' = ').replace(',', ', ') +" )"
 
  # sql = "TRAIN MODEL New_Model MODELTYPE remote_tablegan ON instacart_small.order_products( reordered ) OPTIONS ('epochs' = 1)"
 
+ logger.debug(f'kwargs["options"] : {kwargs["options"]}')
  logger.debug(f'wgkim sql : {sql}')
  return cls._execute(sql)