def __init__(self,
root,
transform=None,
pre_transform=None,
pre_filter=None,
seed=0):
super().__init__(root,
"musae-wiki",
transform,
pre_transform=DigitizeY(bins=[2.5, 3, 3.5, 4, 4.5],
transform_y=np.log10),
pre_filter=self.pre_filter)
mask_init(self, seed=12345 + seed)
def __init__(self, root, name, transform=None, pre_transform=None, seed=0):
super().__init__(root, name, transform, pre_transform)
mask_init(self, seed=12345 + seed)
def __init__(self, root, name, transform=None, pre_transform=None, seed=0):
self.pca_dim = 500
self.name = name
super().__init__(root, name, transform, pre_transform)
mask_init(self, seed=12345 + seed)