File tree Expand file tree Collapse file tree 1 file changed +3
-5
lines changed
chebai/preprocessing/datasets Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Original file line number Diff line number Diff 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 } " )
You can’t perform that action at this time.
0 commit comments