b22ee075 commited on
Commit
2c4cd2e
·
verified ·
1 Parent(s): eed04a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ from sklearn.metrics import mean_absolute_error
8
  import xgboost as xgb
9
  import lightgbm as lgb
10
  from sklearn.ensemble import HistGradientBoostingRegressor
11
- import catboost as cb
12
  import matplotlib.pyplot as plt
13
 
14
  def fetch_data(symbol, start, end):
@@ -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': cb.CatBoostRegressor(verbose=0),
44
  'HistGradientBoosting': HistGradientBoostingRegressor()
45
  }
46
 
 
8
  import xgboost as xgb
9
  import lightgbm as lgb
10
  from sklearn.ensemble import HistGradientBoostingRegressor
11
+ #import catboost as cb
12
  import matplotlib.pyplot as plt
13
 
14
  def fetch_data(symbol, start, end):
 
40
  models = {
41
  'XGBoost': xgb.XGBRegressor(),
42
  'LightGBM': lgb.LGBMRegressor(),
43
+ # 'CatBoost': cb.CatBoostRegressor(verbose=0),
44
  'HistGradientBoosting': HistGradientBoostingRegressor()
45
  }
46