datasets

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,
+        )