(self, trainval_df)
| 869 | |
| 870 | class MyDataset_pairwise(TensorDataset): |
| 871 | def __init__(self, trainval_df): |
| 872 | super(MyDataset_pairwise, self).__init__() |
| 873 | self.df = trainval_df |
| 874 | #self.df.reset_index(drop=True, inplace=True) |
| 875 | def __len__(self): |
| 876 | return len(self.df) |
| 877 |