add predict_proba to base classifier, improve historic predictions handling

2022-08-09 17:31:38 +02:00
parent d36da95941
commit 23cc21ce59
3 changed files with 54 additions and 8 deletions
--- a/freqtrade/freqai/data_drawer.py
+++ b/freqtrade/freqai/data_drawer.py
@@ -358,10 +358,12 @@ class FreqaiDataDrawer:

        dk.find_features(dataframe)

-        if self.freqai_info.get('predict_proba', []):
-            full_labels = dk.label_list + self.freqai_info['predict_proba']
-        else:
-            full_labels = dk.label_list
+        added_labels = []
+        if dk.unique_classes:
+            for label in dk.unique_classes:
+                added_labels += dk.unique_classes[label]
+   
+        full_labels = dk.label_list + added_labels

        for label in full_labels:
            dataframe[label] = 0