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 )