Skip to content

Commit 29e4211

Browse files
0.22.2
1 parent a373c4e commit 29e4211

2 files changed

Lines changed: 5 additions & 1 deletion

File tree

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ build-backend = "setuptools.build_meta"
77

88
[project]
99
name = "spotpython"
10-
version = "0.22.1"
10+
version = "0.22.2"
1111
authors = [
1212
{ name="T. Bartz-Beielstein", email="tbb@bartzundbartz.de" }
1313
]

src/spotpython/data/lightcrossvalidationdatamodule.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,8 @@ class LightCrossValidationDataModule(L.LightningDataModule):
1919
The full training dataset from which training and validation sets will be derived.
2020
data_test (torch.utils.data.Dataset, optional):
2121
The separate test dataset that will be used for testing.
22+
data_val (torch.utils.data.Dataset, optional):
23+
The separate validation dataset that will be used for validation.
2224
k (int): The fold number. Defaults to 1.
2325
split_seed (int): The random seed for splitting the data. Defaults to 42.
2426
num_splits (int): The number of splits for cross-validation. Defaults to 10.
@@ -49,6 +51,7 @@ def __init__(
4951
dataset: Optional[object] = None,
5052
data_full_train: Optional[object] = None,
5153
data_test: Optional[object] = None,
54+
data_val: Optional[object] = None,
5255
k: int = 1,
5356
split_seed: int = 42,
5457
num_splits: int = 10,
@@ -63,6 +66,7 @@ def __init__(
6366
self.data_full = dataset
6467
self.data_full_train = data_full_train
6568
self.data_test = data_test
69+
self.data_val = data_val
6670
self.data_dir = data_dir
6771
self.num_workers = num_workers
6872
self.k = k

0 commit comments

Comments
 (0)