fix bug in return_values()

This commit is contained in:
robcaulk 2022-06-03 16:58:51 +02:00
parent 16b4a5b71f
commit f2762e3b4b

View File

@ -24,7 +24,7 @@ class CatboostPredictionModel(IFreqaiModel):
dataframe["do_predict"] = dh.full_do_predict dataframe["do_predict"] = dh.full_do_predict
dataframe["target_mean"] = dh.full_target_mean dataframe["target_mean"] = dh.full_target_mean
dataframe["target_std"] = dh.full_target_std dataframe["target_std"] = dh.full_target_std
if self.freqai_info('feature_parameters', {}).get('DI-threshold', 0) > 0: if self.freqai_info.get('feature_parameters', {}).get('DI-threshold', 0) > 0:
dataframe["DI"] = dh.full_DI_values dataframe["DI"] = dh.full_DI_values
return dataframe return dataframe