ensure predict_proba follows suit. Remove all lib specific params from example config

2022-12-11 12:39:31 +01:00
parent 85f22b5c30
commit 8c7ec07951
2 changed files with 2 additions and 1 deletions
--- a/freqtrade/freqai/base_models/BaseClassifierModel.py
+++ b/freqtrade/freqai/base_models/BaseClassifierModel.py
@@ -101,6 +101,8 @@ class BaseClassifierModel(IFreqaiModel):
        pred_df = DataFrame(predictions, columns=dk.label_list)

        predictions_prob = self.model.predict_proba(dk.data_dictionary["prediction_features"])
+        if self.CONV_WIDTH == 1:
+            predictions_prob = np.reshape(predictions_prob, (-1, len(self.model.classes_)))
        pred_df_prob = DataFrame(predictions_prob, columns=self.model.classes_)

        pred_df = pd.concat([pred_df, pred_df_prob], axis=1)