Hi,
Try to run this line out of the dataloader and see what its shape is. Obviously, the issue is the way numericals is constructed as dataloader only create a batch of that data which as you have mentioned, [10, ..., ...] corresponds to batch size you have defined.



 Jordan_Howell:

numericals = np.asarray(self.df.loc[idx, self.numerical_columns].values)