def __init__(
self,
batch_size: int,
n_workers: int,
train_val_split: float = 0.9,
test_split: float = 0.1,
shifting_proportion_total: float = 0.1,
shifting_proportion_k: float = 1e-2,
**kwargs,
):
UCIKin8nmDataModule.__init__(
self,
batch_size,
n_workers,
train_val_split,
test_split,
)
DataModuleShifted.__init__(self, shifting_proportion_total,
shifting_proportion_k)
def __init__(
self,
batch_size: int,
n_workers: int,
N_train: int = 2000,
N_test: int = 1000,
train_val_split: float = 0.9,
shifting_proportion_total: float = 0.1,
shifting_proportion_k: float = 1e-2,
**kwargs,
):
Toy2DDataModule.__init__(
self,
batch_size,
n_workers,
N_train,
N_test,
train_val_split,
)
DataModuleShifted.__init__(
self, shifting_proportion_total, shifting_proportion_k
)
def setup(self, stage: str = None):
UCICCPPDataModule.setup(self, stage)
DataModuleShifted.setup(self)
def setup(self, stage: str = None):
UCIKin8nmDataModule.setup(self, stage)
DataModuleShifted.setup(self)
def setup(self, stage: str = None):
UCIBostonDataModule.setup(self, stage)
DataModuleShifted.setup(self)
def setup(self, stage: str = None):
UCIWineWhiteDataModule.setup(self, stage)
DataModuleShifted.setup(self)
def setup(self, stage: str = None):
Toy2DDataModule.setup(self, stage)
DataModuleShifted.setup(self)
def setup(self, stage: str = None):
UCIEnergyDataModule.setup(self, stage)
DataModuleShifted.setup(self)
def setup(self, stage: str = None):
UCISuperConductDataModule.setup(self, stage)
DataModuleShifted.setup(self)
def setup(self, stage: str = None):
UCIProteinDataModule.setup(self, stage)
DataModuleShifted.setup(self)