Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
| 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 |
|