Skip to content

Commit 8f369bf

Browse files
committed
validation data option for lightdatamodule
1 parent c2c172e commit 8f369bf

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

src/spotpython/data/lightdatamodule.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -217,7 +217,7 @@ def _setup_test_data_provided(self, stage) -> None:
217217
# Transform the predict data
218218
self.data_predict = self.transform_dataset(self.data_predict)
219219

220-
def _setup_data_val_provided(self, stage) -> None:
220+
def _setup_val_data_provided(self, stage) -> None:
221221
# New functionality for predefined train, validation and test data in the fun_control
222222
# Get the data set sizes
223223
if self.data_full_train is None:
@@ -229,7 +229,6 @@ def _setup_data_val_provided(self, stage) -> None:
229229
if stage == "fit" or stage is None:
230230
if self.verbosity > 0:
231231
print(f"train_size: {train_size}, val_size: {val_size} used for train & val data.")
232-
generator_fit = torch.Generator().manual_seed(self.test_seed)
233232
# Use all data from data_full_train as training data
234233
self.data_train = self.data_full_train
235234
# Handle scaling and transformation if scaler is provided
@@ -281,7 +280,7 @@ def setup(self, stage: Optional[str] = None) -> None:
281280
if self.data_full is not None:
282281
self._setup_full_data_provided(stage)
283282
elif self.data_val is not None:
284-
self._setup_data_val_provided(stage)
283+
self._setup_val_data_provided(stage)
285284
else:
286285
self._setup_test_data_provided(stage)
287286

0 commit comments

Comments
 (0)