Method __init__

pairwise/dataloader.py:261–264

(self, trainval_df)

class MyDataset(TensorDataset):
    def __init__(self, trainval_df):
        super(MyDataset, self).__init__()
        self.df = trainval_df
        self.df.reset_index(drop=True, inplace=True)  # train_test_split之后，数据集的index混乱，需要reset

    def __len__(self):
        return len(self.df)