From 4d0226f3636f536a089e1ae6c2a50d19fc51bffa Mon Sep 17 00:00:00 2001
From: Paul Andrey <paul.andrey@inria.fr>
Date: Wed, 30 Aug 2023 12:19:37 +0200
Subject: [PATCH] Fix 'InMemoryDataset' single-column target loading from csv.
---
declearn/dataset/_inmemory.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/declearn/dataset/_inmemory.py b/declearn/dataset/_inmemory.py
index 7f3b33fa..f8e5f54d 100644
--- a/declearn/dataset/_inmemory.py
+++ b/declearn/dataset/_inmemory.py
@@ -149,6 +149,11 @@ class InMemoryDataset(Dataset):
target = self.data[target]
else:
target = load_data_array(target)
+ if (
+ isinstance(target, pd.DataFrame)
+ and len(target.columns) == 1
+ ):
+ target = target.iloc[:, 0]
self.target = target
# Assign the (optional) sample weights data array.
if isinstance(s_wght, str):
--
GitLab