(self, trainval_df)
| 647 | |
| 648 | class MyDataset_trans(TensorDataset): |
| 649 | def __init__(self, trainval_df): |
| 650 | super(MyDataset_trans, self).__init__() |
| 651 | self.df = trainval_df |
| 652 | #self.df.reset_index(drop=True, inplace=True) |
| 653 | def __len__(self): |
| 654 | return len(self.df) |
| 655 |