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
)