Skip to content

Commit af1295b

Browse files
0.19.3
1 parent 6b7f0a1 commit af1295b

2 files changed

Lines changed: 7 additions & 2 deletions

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.19.2"
10+
version = "0.19.3"
1111
authors = [
1212
{ name="T. Bartz-Beielstein", email="tbb@bartzundbartz.de" }
1313
]

src/spotpython/utils/init.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ def fun_control_init(
3030
core_model_name=None,
3131
data=None,
3232
data_full_train=None,
33+
data_val=None,
3334
data_dir="./data",
3435
data_module=None,
3536
data_set=None,
@@ -128,8 +129,11 @@ def fun_control_init(
128129
data_dir (str):
129130
The directory to save the data. Default is "./data".
130131
data_full_train (torch.utils.data.Dataset, optional):
131-
The full training dataset from which training and validation sets will be derived.
132+
The full training dataset from which training and validation sets will be derived if data_val is None.
132133
Default is None.
134+
data_val (torch.utils.data.Dataset, optional):
135+
The validation dataset. Default is None. If not None, the training and validation sets are derived from
136+
the full training dataset (data_full_train) and the validation dataset (data_val).
133137
data_module (object):
134138
The data module object. Default is None.
135139
data_set (object):
@@ -399,6 +403,7 @@ def fun_control_init(
399403
"data_set": data_set,
400404
"data_set_name": data_set_name,
401405
"data_test": data_test,
406+
"data_val": data_val,
402407
"db_dict_name": db_dict_name,
403408
"design": design,
404409
"device": device,

0 commit comments

Comments
 (0)