From 443263803ce332bc6aed38a1502594a31af9c49a Mon Sep 17 00:00:00 2001 From: Yinon Polak Date: Tue, 21 Mar 2023 11:42:05 +0200 Subject: [PATCH] unsqueeze target tensor when 1 dimensional --- 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 8097b8b85..52e6d5138 100644 --- a/freqtrade/freqai/base_models/PyTorchModelTrainer.py +++ b/freqtrade/freqai/base_models/PyTorchModelTrainer.py @@ -124,7 +124,7 @@ class PyTorchModelTrainer: data_loader_dictionary = {} for split in ["train", "test"]: labels_shape = data_dictionary[f"{split}_labels"].shape - labels_view = labels_shape[0] if labels_shape[1] == 1 else labels_shape + labels_view = (labels_shape[0], 1) if labels_shape[1] == 1 else labels_shape dataset = TensorDataset( torch.from_numpy(data_dictionary[f"{split}_features"].values).float(), torch.from_numpy(data_dictionary[f"{split}_labels"].values)