fix bug in return_values()
This commit is contained in:
parent
16b4a5b71f
commit
f2762e3b4b
@ -24,7 +24,7 @@ class CatboostPredictionModel(IFreqaiModel):
|
||||
dataframe["do_predict"] = dh.full_do_predict
|
||||
dataframe["target_mean"] = dh.full_target_mean
|
||||
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
|
||||
|
||||
return dataframe
|
||||
|
Loading…
Reference in New Issue
Block a user