@@ -47,14 +47,23 @@ def __init__(
47
47
"""TrainingSettings class for Azure Machine Learning.
48
48
49
49
:param enable_onnx_compatible_models: If set to True, the model will be trained to be compatible with ONNX
50
+ :type enable_onnx_compatible_models: typing.Optional[bool]
50
51
:param enable_dnn_training: If set to True,the model will use DNN training
52
+ :type enable_dnn_training: typing.Optional[bool]
51
53
:param enable_model_explainability: If set to True, the model will be trained to be explainable
54
+ :type enable_model_explainability: typing.Optional[bool]
52
55
:param enable_stack_ensemble: If set to True, a final ensemble model will be created using a stack of models
56
+ :type enable_stack_ensemble: typing.Optional[bool]
53
57
:param enable_vote_ensemble: If set to True, a final ensemble model will be created using a voting ensemble
58
+ :type enable_vote_ensemble: typing.Optional[bool]
54
59
:param stack_ensemble_settings: Settings for stack ensemble
60
+ :type stack_ensemble_settings: typing.Optional[azure.ai.ml.automl.StackEnsembleSettings]
55
61
:param ensemble_model_download_timeout: Timeout for downloading ensemble models
62
+ :type ensemble_model_download_timeout: typing.Optional[typing.List[int]]
56
63
:param allowed_training_algorithms: Models to train
64
+ :type allowed_training_algorithms: typing.Optional[typing.List[str]]
57
65
:param blocked_training_algorithms: Models that will not be considered for training
66
+ :type blocked_training_algorithms: typing.Optional[typing.List[str]]
58
67
:param training_mode: [Experimental] The training mode to use.
59
68
The possible values are-
60
69
@@ -65,6 +74,7 @@ def __init__(
65
74
* auto- Currently, it is same as non_distributed. In future, this might change.
66
75
67
76
Note: This parameter is in public preview and may change in future.
77
+ :type training_mode: typing.Optional[typing.Union[str, azure.ai.ml.constants.TabularTrainingMode]]
68
78
"""
69
79
self .enable_onnx_compatible_models = enable_onnx_compatible_models
70
80
self .enable_dnn_training = enable_dnn_training
@@ -79,7 +89,7 @@ def __init__(
79
89
80
90
@experimental
81
91
@property
82
- def training_mode (self ):
92
+ def training_mode (self ) -> Optional [ TabularTrainingMode ] :
83
93
return self ._training_mode
84
94
85
95
@training_mode .setter
@@ -102,11 +112,11 @@ def training_mode(self, value: Optional[Union[str, TabularTrainingMode]]):
102
112
)
103
113
104
114
@property
105
- def allowed_training_algorithms (self ):
115
+ def allowed_training_algorithms (self ) -> Optional [ List [ str ]] :
106
116
return self ._allowed_training_algorithms
107
117
108
118
@property
109
- def blocked_training_algorithms (self ):
119
+ def blocked_training_algorithms (self ) -> Optional [ List [ str ]] :
110
120
return self ._blocked_training_algorithms
111
121
112
122
def _to_rest_object (self ) -> RestTrainingSettings :
0 commit comments