From 348a08f1c41b47601bb5592280a567e7c0225b8b Mon Sep 17 00:00:00 2001 From: Yinon Polak Date: Mon, 6 Mar 2023 16:41:47 +0200 Subject: [PATCH] add todo - currently assuming class labels are strings ['0.0', '1.0' .. n_classes]. need to resolve it per ClassifierModel --- freqtrade/freqai/base_models/PyTorchModelTrainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/freqtrade/freqai/base_models/PyTorchModelTrainer.py b/freqtrade/freqai/base_models/PyTorchModelTrainer.py index 13c5ffe74..03d264371 100644 --- a/freqtrade/freqai/base_models/PyTorchModelTrainer.py +++ b/freqtrade/freqai/base_models/PyTorchModelTrainer.py @@ -101,7 +101,7 @@ class PyTorchModelTrainer: torch.from_numpy(data_dictionary[f'{split}_features'].values).float(), torch.from_numpy(data_dictionary[f'{split}_labels'].astype(float).values) .long() - .view(labels_view) + .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,