debug classifier with predict proba

2022-08-10 15:16:50 +02:00
parent 23cc21ce59
commit b1b76a2dbe
4 changed files with 21 additions and 24 deletions
--- a/freqtrade/freqai/data_drawer.py
+++ b/freqtrade/freqai/data_drawer.py
@@ -358,12 +358,7 @@ class FreqaiDataDrawer:

        dk.find_features(dataframe)

-        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
+        full_labels = dk.label_list + dk.unique_class_list

        for label in full_labels:
            dataframe[label] = 0