Skip to content

Commit a8668b8

Browse files
bartzbeielsteinbartzbeielstein
committed
examples formatted
1 parent 9f9c6b2 commit a8668b8

2 files changed

Lines changed: 37 additions & 37 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.6.45"
10+
version = "0.6.46"
1111
authors = [
1212
{ name="T. Bartz-Beielstein", email="tbb@bartzundbartz.de" }
1313
]

‎src/spotPython/data/lightdatamodule.py‎

Lines changed: 36 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -61,15 +61,15 @@ def setup(self, stage: Optional[str] = None) -> None:
6161
stage (Optional[str]): The current stage. Defaults to None.
6262
6363
Examples:
64-
>>> from spotPython.data.lightdatamodule import LightDataModule
65-
from spotPython.data.csvdataset import CSVDataset
66-
from spotPython.data.pkldataset import PKLDataset
67-
import torch
68-
dataset = CSVDataset(csv_file='data.csv', target_column='prognosis', feature_type=torch.long)
69-
data_module = LightDataModule(dataset=dataset, batch_size=5, test_size=0.5)
70-
data_module.setup()
71-
print(f"Training set size: {len(data_module.data_train)}")
72-
Training set size: 3
64+
>>> from spotPython.data.lightdatamodule import LightDataModule
65+
from spotPython.data.csvdataset import CSVDataset
66+
from spotPython.data.pkldataset import PKLDataset
67+
import torch
68+
dataset = CSVDataset(csv_file='data.csv', target_column='prognosis', feature_type=torch.long)
69+
data_module = LightDataModule(dataset=dataset, batch_size=5, test_size=0.5)
70+
data_module.setup()
71+
print(f"Training set size: {len(data_module.data_train)}")
72+
Training set size: 3
7373
7474
"""
7575
# if test_size is float, then train_size is 1 - test_size
@@ -107,15 +107,15 @@ def train_dataloader(self) -> DataLoader:
107107
DataLoader: The training dataloader.
108108
109109
Examples:
110-
>>> from spotPython.data.lightdatamodule import LightDataModule
111-
from spotPython.data.csvdataset import CSVDataset
112-
from spotPython.data.pkldataset import PKLDataset
113-
import torch
114-
dataset = CSVDataset(csv_file='data.csv', target_column='prognosis', feature_type=torch.long)
115-
data_module = LightDataModule(dataset=dataset, batch_size=5, test_size=0.5)
116-
data_module.setup()
117-
print(f"Training set size: {len(data_module.data_train)}")
118-
Training set size: 3
110+
>>> from spotPython.data.lightdatamodule import LightDataModule
111+
from spotPython.data.csvdataset import CSVDataset
112+
from spotPython.data.pkldataset import PKLDataset
113+
import torch
114+
dataset = CSVDataset(csv_file='data.csv', target_column='prognosis', feature_type=torch.long)
115+
data_module = LightDataModule(dataset=dataset, batch_size=5, test_size=0.5)
116+
data_module.setup()
117+
print(f"Training set size: {len(data_module.data_train)}")
118+
Training set size: 3
119119
120120
"""
121121
return DataLoader(self.data_train, batch_size=self.batch_size, num_workers=self.num_workers)
@@ -128,15 +128,15 @@ def val_dataloader(self) -> DataLoader:
128128
DataLoader: The validation dataloader.
129129
130130
Examples:
131-
>>> from spotPython.data.lightdatamodule import LightDataModule
132-
from spotPython.data.csvdataset import CSVDataset
133-
from spotPython.data.pkldataset import PKLDataset
134-
import torch
135-
dataset = CSVDataset(csv_file='data.csv', target_column='prognosis', feature_type=torch.long)
136-
data_module = LightDataModule(dataset=dataset, batch_size=5, test_size=0.5)
137-
data_module.setup()
138-
print(f"Training set size: {len(data_module.data_val)}")
139-
Training set size: 3
131+
>>> from spotPython.data.lightdatamodule import LightDataModule
132+
from spotPython.data.csvdataset import CSVDataset
133+
from spotPython.data.pkldataset import PKLDataset
134+
import torch
135+
dataset = CSVDataset(csv_file='data.csv', target_column='prognosis', feature_type=torch.long)
136+
data_module = LightDataModule(dataset=dataset, batch_size=5, test_size=0.5)
137+
data_module.setup()
138+
print(f"Training set size: {len(data_module.data_val)}")
139+
Training set size: 3
140140
141141
"""
142142
return DataLoader(self.data_val, batch_size=self.batch_size, num_workers=self.num_workers)
@@ -149,15 +149,15 @@ def test_dataloader(self) -> DataLoader:
149149
DataLoader: The test dataloader.
150150
151151
Examples:
152-
>>> from spotPython.data.lightdatamodule import LightDataModule
153-
from spotPython.data.csvdataset import CSVDataset
154-
from spotPython.data.pkldataset import PKLDataset
155-
import torch
156-
dataset = CSVDataset(csv_file='data.csv', target_column='prognosis', feature_type=torch.long)
157-
data_module = LightDataModule(dataset=dataset, batch_size=5, test_size=0.5)
158-
data_module.setup()
159-
print(f"Test set size: {len(data_module.data_test)}")
160-
Test set size: 6
152+
>>> from spotPython.data.lightdatamodule import LightDataModule
153+
from spotPython.data.csvdataset import CSVDataset
154+
from spotPython.data.pkldataset import PKLDataset
155+
import torch
156+
dataset = CSVDataset(csv_file='data.csv', target_column='prognosis', feature_type=torch.long)
157+
data_module = LightDataModule(dataset=dataset, batch_size=5, test_size=0.5)
158+
data_module.setup()
159+
print(f"Test set size: {len(data_module.data_test)}")
160+
Test set size: 6
161161
162162
"""
163163
return DataLoader(self.data_test, batch_size=self.batch_size, num_workers=self.num_workers)

0 commit comments

Comments
 (0)