change documentation and small bugfix

2023-03-08 15:38:22 +02:00
parent 76fbec0c17
commit 1805db2b07
2 changed files with 13 additions and 11 deletions
--- a/freqtrade/freqai/base_models/PyTorchModelTrainer.py
+++ b/freqtrade/freqai/base_models/PyTorchModelTrainer.py
@@ -104,8 +104,6 @@ class PyTorchModelTrainer:
                .long()
                .view(labels_view)
            )
-            # todo currently assuming class labels are strings ['0.0', '1.0' .. n_classes].
-            #  need to resolve it per ClassifierModel

            data_loader = DataLoader(
                dataset,