|
14 | 14 | - Diptesh |
15 | 15 | - Madhu |
16 | 16 |
|
17 | | - Date: Jan 10, 2022 |
| 17 | + Date: Jan 15, 2022 |
18 | 18 | """ |
19 | 19 |
|
20 | 20 | # pylint: disable=invalid-name |
@@ -110,7 +110,7 @@ def _compute_metrics(self): |
110 | 110 | self.model_summary = model_summary |
111 | 111 |
|
112 | 112 | def _ts_data_transform(self): |
113 | | - """Transform input data with significant lag variables. """ |
| 113 | + """Transform input data with significant lag variables.""" |
114 | 114 | # Determine seasonality and return seaonal lag |
115 | 115 | decomposition = seasonal_decompose(self.df[self.y_var], |
116 | 116 | model="additive") |
@@ -172,7 +172,7 @@ def _ts_predict(self, |
172 | 172 | df_pred_x = pd.DataFrame(df_op.iloc[i]).T |
173 | 173 | for j, _ in enumerate(self.ts_param["ts_lag_var"]): |
174 | 174 | df_pred_x["lag_" + str(self.ts_param["ts_lag_var"][j])] \ |
175 | | - = lst_lag_val[len(lst_lag_val) \ |
| 175 | + = lst_lag_val[len(lst_lag_val) |
176 | 176 | - self.ts_param["ts_lag_var"][j]] |
177 | 177 | df_pred_x = pd.DataFrame(df_pred_x) |
178 | 178 | y_hat = self.model.predict(df_pred_x[self.ts_param["ts_x_var"]]) |
@@ -224,10 +224,6 @@ class RandomForest(Tree): |
224 | 224 |
|
225 | 225 | Number of cross validations folds (the default is 5) |
226 | 226 |
|
227 | | - max_lag : int, optional |
228 | | -
|
229 | | - Maximum lag value for auto regression in timeseries (the default is 20) |
230 | | -
|
231 | 227 | threshold : float, optional |
232 | 228 |
|
233 | 229 | Threshold to identify significant lag values (the default is 0.05) |
@@ -357,10 +353,6 @@ class XGBoost(Tree): |
357 | 353 |
|
358 | 354 | Number of cross validations folds (the default is 5) |
359 | 355 |
|
360 | | - max_lag : int, optional |
361 | | -
|
362 | | - Maximum lag value for auto regression in timeseries (the default is 20) |
363 | | -
|
364 | 356 | threshold : float, optional |
365 | 357 |
|
366 | 358 | Threshold to identify significant lag values (the default is 0.05) |
|
0 commit comments