From 3dde1a3358758bf9bd5ebf8bb9bc284c07a4c5b6 Mon Sep 17 00:00:00 2001 From: henrykironde Date: Tue, 19 Nov 2024 22:19:02 -0500 Subject: [PATCH] Add template for milliontreesUnlabeledDataset --- .../unlabeled/milliontrees_unlabeled_dataset.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 milliontrees/datasets/unlabeled/milliontrees_unlabeled_dataset.py diff --git a/milliontrees/datasets/unlabeled/milliontrees_unlabeled_dataset.py b/milliontrees/datasets/unlabeled/milliontrees_unlabeled_dataset.py new file mode 100644 index 0000000..624cbf3 --- /dev/null +++ b/milliontrees/datasets/unlabeled/milliontrees_unlabeled_dataset.py @@ -0,0 +1,16 @@ +from torch.utils.data import Dataset + +class milliontreesUnlabeledDataset(Dataset): + """Base class for unlabeled MillionTrees datasets.""" + + def __init__(self, root_dir, download, split_scheme): + self.root_dir = root_dir + + def initialize_data_dir(self, root_dir, download): + """Initialize the data directory.""" + return root_dir + + @property + def data_dir(self): + """The path to the data directory.""" + return self._data_dir \ No newline at end of file