Skip to content

Commit 4043b23

Browse files
committed
skip metadata calculation
1 parent feea315 commit 4043b23

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

chebai/preprocessing/datasets/pubchem.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -185,16 +185,14 @@ def processed_file_names(self) -> List[str]:
185185

186186
def _set_processed_data_props(self):
187187
"""
188-
Load processed data and extract metadata.
188+
Self-supervised learning with PubChem does not use this metadata, therefore set them to zero.
189189
190190
Sets:
191-
- self._num_of_labels: Number of target labels in the dataset.
191+
- self._num_of_labels: 0
192192
- self._feature_vector_size: 0.
193193
"""
194-
with open(os.path.join(self.processed_dir_main, "classes.txt")) as f:
195-
classes = [f.strip() for f in f.readlines() if f.strip()]
196194

197-
self._num_of_labels = len(classes)
195+
self._num_of_labels = 0
198196
self._feature_vector_size = 0
199197

200198
print(f"Number of labels for loaded data: {self._num_of_labels}")

0 commit comments

Comments
 (0)