Exemplos de DataModuleShifted.__init__ em Python

Exemplo n.º 1

Arquivo: datamodule.py Projeto: pierresegonne/SGGM

 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)

Exemplo n.º 2

 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
     )