Update BaseClassifierModel.py

This commit is contained in:
Robert Caulk
2023-06-18 11:31:03 +02:00
committed by GitHub
parent 7e2f857aa5
commit cca4fa1178

View File

@@ -117,7 +117,7 @@ class BaseClassifierModel(IFreqaiModel):
pred_df = pd.concat([pred_df, pred_df_prob], axis=1)
if self.freqai_info.get("DI_threshold", 0) > 0:
if dk.feature_pipeline["di"]:
dk.DI_values = dk.feature_pipeline["di"].di_values
else:
dk.DI_values = np.zeros(len(outliers.index))