Skip to content

Commit d7d05c8

Browse files
nabenabe0928ravinkohli
authored andcommitted
[doc] Add the reference to the fit_dictionary
1 parent 03ac94a commit d7d05c8

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

341341
self.indices[fold] = ((train_indices, test_indices))
342342

343+
# See autoPyTorch/pipeline/components/base_component.py::autoPyTorchComponent for more details
344+
# about fit_dictionary
343345
X = {'train_indices': train_indices,
344346
'val_indices': test_indices,
345347
'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
@@ -136,6 +136,8 @@ def get_device_from_fit_dictionary(X: Dict[str, Any]) -> torch.device:
136136
137137
Args:
138138
X (Dict[str, Any]): A fit dictionary to control how the pipeline is fitted
139+
See autoPyTorch/pipeline/components/base_component.py::autoPyTorchComponent for more details
140+
about fit_dictionary
139141
140142
Returns:
141143
torch.device: Device to be used for training/inference

0 commit comments

Comments
 (0)