Creating custom pytorch dataset class for n number of target columns

Last few days I was googling about this issue which I created here , finally this post helped me in resolving my problem . Thanks :blush: