diff --git a/src/beignet/datasets/_chen_sabdab_dataset.py b/src/beignet/datasets/_chen_sabdab_dataset.py new file mode 100644 index 0000000000..7d438e75b3 --- /dev/null +++ b/src/beignet/datasets/_chen_sabdab_dataset.py @@ -0,0 +1,44 @@ +from pathlib import Path +from typing import Callable + +from beignet.transforms import Transform + +from ._tdc_dataset import TDCDataset + + +class ChenSAbDabDataset(TDCDataset): + def __init__( + self, + root: str | Path, + *, + download: bool = False, + transform: Callable | Transform | None = None, + target_transform: Callable | Transform | None = None, + ): + r""" + Parameters + ---------- + root : str | Path + Root directory of dataset. + + download: bool + If `True`, downloads the dataset to the root directory. If dataset + already exists, it is not redownloaded. Default, `False`. + + transform : Callable | Transform | None + Transforms the input. + + target_transform : Callable | Transform | None + Transforms the target. + """ + super().__init__( + root=root, + download=download, + identifier=4167164, + suffix="tsv", + checksum="md5:3b58b73e0d5371a2ccc865ca045a2061", + x_keys=["X"], + y_keys=["Y"], + transform=transform, + target_transform=target_transform, + )