Skip to content

Commit 50983f1

Browse files
nabenabe0928ravinkohli
authored andcommitted
[doc] Add the reference to the fit_dictionary
1 parent eebbd36 commit 50983f1

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
@@ -356,6 +356,8 @@ def _fit_and_predict(self, pipeline: BaseEstimator, fold: int, train_indices: Un
356356

357357
self.indices[fold] = ((train_indices, test_indices))
358358

359+
# See autoPyTorch/pipeline/components/base_component.py::autoPyTorchComponent for more details
360+
# about fit_dictionary
359361
X = {'train_indices': train_indices,
360362
'val_indices': test_indices,
361363
'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
@@ -158,6 +158,8 @@ def get_device_from_fit_dictionary(X: Dict[str, Any]) -> torch.device:
158158
159159
Args:
160160
X (Dict[str, Any]): A fit dictionary to control how the pipeline is fitted
161+
See autoPyTorch/pipeline/components/base_component.py::autoPyTorchComponent for more details
162+
about fit_dictionary
161163
162164
Returns:
163165
torch.device: Device to be used for training/inference

0 commit comments

Comments
 (0)