b22ee075 commited on
Commit
b11d2e6
·
verified ·
1 Parent(s): 7a345a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def train_predict(symbol, start_date, end_date, forecast_days):
40
  models = {
41
  'XGBoost': xgb.XGBRegressor(),
42
  'LightGBM': lgb.LGBMRegressor(),
43
- 'CatBoost': catboost.CatBoostRegressor(),
44
  'HistGradientBoosting': HistGradientBoostingRegressor()
45
  }
46
 
 
40
  models = {
41
  'XGBoost': xgb.XGBRegressor(),
42
  'LightGBM': lgb.LGBMRegressor(),
43
+ # 'CatBoost': catboost.CatBoostRegressor(),
44
  'HistGradientBoosting': HistGradientBoostingRegressor()
45
  }
46