fix: use .shape instead of index for outliers

This commit is contained in:
robcaulk
2023-06-25 16:34:44 +02:00
parent 9da28e5328
commit fca73531cf
7 changed files with 7 additions and 7 deletions

View File

@@ -800,7 +800,7 @@ class MyCoolFreqaiModel(BaseRegressionModel):
if self.freqai_info.get("DI_threshold", 0) > 0:
dk.DI_values = dk.feature_pipeline["di"].di_values
else:
dk.DI_values = np.zeros(len(outliers.index))
dk.DI_values = np.zeros(outliers.shape[0])
dk.do_predict = outliers
# ... your custom code