From c9af163c4bdad378b2c202feacb7559fc2912c9a Mon Sep 17 00:00:00 2001
From: pswain <peter.swain@ed.ac.uk>
Date: Fri, 24 Nov 2023 16:00:32 +0000
Subject: [PATCH] feat(dataloader): prints shape of data frame created
---
dataloader.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/dataloader.py b/dataloader.py
index edbb7f8..2ccd88c 100644
--- a/dataloader.py
+++ b/dataloader.py
@@ -255,6 +255,7 @@ class dataloader:
self.df = pd.merge(self.df, r_df, how="left")
else:
self.df = r_df
+ print(f" data size is {self.df.shape}")
# define ids
self.ids = list(self.df.id.unique())
if not use_tsv:
--
GitLab