def test_dataloader(self):
return DataLoader(
SklearnDataset(self.x_test, self.y_test, self._x_type, self._y_type), batch_size=self.batch_size
)
def val_dataloader(self):
return DataLoader(
SklearnDataset(self.x_valid, self.y_valid, self._x_type, self._y_type), batch_size=self.batch_size
)