コード例 #1
0
 def test_dataloader(self):
     return DataLoader(
         SklearnDataset(self.x_test, self.y_test, self._x_type, self._y_type), batch_size=self.batch_size
     )
コード例 #2
0
 def val_dataloader(self):
     return DataLoader(
         SklearnDataset(self.x_valid, self.y_valid, self._x_type, self._y_type), batch_size=self.batch_size
     )