Skip to content

Commit ebb9cf6

Browse files
nabenabe0928ravinkohli
authored andcommitted
[doc] Add the reference to the fit_dictionary
1 parent a4ba934 commit ebb9cf6

File tree

3 files changed

+8
-0
lines changed

3 files changed

+8
-0
lines changed

autoPyTorch/evaluation/train_evaluator.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -362,6 +362,8 @@ def _fit_and_predict(self, pipeline: BaseEstimator, fold: int, train_indices: Un
362362

363363
self.indices[fold] = ((train_indices, test_indices))
364364

365+
# See autoPyTorch/pipeline/components/base_component.py::autoPyTorchComponent for more details
366+
# about fit_dictionary
365367
X = {'train_indices': train_indices,
366368
'val_indices': test_indices,
367369
'split_id': fold,

autoPyTorch/pipeline/components/preprocessing/tabular_preprocessing/utils.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,12 @@ def get_tabular_preprocessers(X: Dict[str, Any]) -> Dict[str, List[BaseEstimator
1212
Creates a dictionary with two keys,
1313
numerical- containing list of numerical preprocessors
1414
categorical- containing list of categorical preprocessors
15+
1516
Args:
1617
X: fit dictionary
18+
See autoPyTorch/pipeline/components/base_component.py::autoPyTorchComponent for more details
19+
about fit_dictionary
20+
1721
Returns:
1822
(Dict[str, List[BaseEstimator]]): dictionary with list of numerical and categorical preprocessors
1923
"""

autoPyTorch/utils/common.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -166,6 +166,8 @@ def get_device_from_fit_dictionary(X: Dict[str, Any]) -> torch.device:
166166
167167
Args:
168168
X (Dict[str, Any]): A fit dictionary to control how the pipeline is fitted
169+
See autoPyTorch/pipeline/components/base_component.py::autoPyTorchComponent for more details
170+
about fit_dictionary
169171
170172
Returns:
171173
torch.device: Device to be used for training/inference

0 commit comments

Comments
 (0)