@@ -76,7 +76,7 @@ def __init__(
76
76
ensemble_size : int, optional (default=50)
77
77
Number of models added to the ensemble built by *Ensemble
78
78
selection from libraries of models*. Models are drawn with
79
- replacement.
79
+ replacement. If set to ``0`` no ensemble is fit.
80
80
81
81
ensemble_nbest : int, optional (default=50)
82
82
Only consider the ``ensemble_nbest`` models when building an
@@ -96,10 +96,14 @@ def __init__(
96
96
memory_limit : int, optional (3072)
97
97
Memory limit in MB for the machine learning algorithm.
98
98
`auto-sklearn` will stop fitting the machine learning algorithm if
99
- it tries to allocate more than `memory_limit` MB.
100
- If None is provided, no memory limit is set.
101
- In case of multi-processing, `memory_limit` will be per job.
102
- This memory limit also applies to the ensemble creation process.
99
+ it tries to allocate more than ``memory_limit`` MB.
100
+
101
+ **Important notes:**
102
+
103
+ * If ``None`` is provided, no memory limit is set.
104
+ * In case of multi-processing, ``memory_limit`` will be *per job*, so the total usage is
105
+ ``n_jobs x memory_limit``.
106
+ * The memory limit also applies to the ensemble creation process.
103
107
104
108
include : dict, optional (None)
105
109
If None, all possible algorithms are used. Otherwise specifies
@@ -145,10 +149,10 @@ def __init__(
145
149
* 'cv-iterative-fit': {'folds': int}
146
150
* 'partial-cv': {'folds': int, 'shuffle': bool}
147
151
* BaseCrossValidator or _RepeatedSplits or BaseShuffleSplit object: all arguments
148
- required by chosen class as specified in scikit-learn documentation.
149
- If arguments are not provided, scikit-learn defaults are used.
150
- If no defaults are available, an exception is raised.
151
- Refer to the 'n_splits' argument as 'folds'.
152
+ required by chosen class as specified in scikit-learn documentation.
153
+ If arguments are not provided, scikit-learn defaults are used.
154
+ If no defaults are available, an exception is raised.
155
+ Refer to the 'n_splits' argument as 'folds'.
152
156
153
157
tmp_folder : string, optional (None)
154
158
folder to store configuration output and log files, if ``None``
@@ -160,13 +164,15 @@ def __init__(
160
164
161
165
n_jobs : int, optional, experimental
162
166
The number of jobs to run in parallel for ``fit()``. ``-1`` means
163
- using all processors. By default, Auto-sklearn uses a single core
164
- for fitting the machine learning model and a single core for fitting
165
- an ensemble. Ensemble building is not affected by ``n_jobs`` but
166
- can be controlled by the number of models in the ensemble. In
167
- contrast to most scikit-learn models, ``n_jobs`` given in the
168
- constructor is not applied to the ``predict()`` method. If
169
- ``dask_client`` is None, a new dask client is created.
167
+ using all processors.
168
+
169
+ **Important notes**:
170
+
171
+ * By default, Auto-sklearn uses one core.
172
+ * Ensemble building is not affected by ``n_jobs`` but can be controlled by the number
173
+ of models in the ensemble.
174
+ * ``predict()`` is not affected by ``n_jobs`` (in contrast to most scikit-learn models)
175
+ * If ``dask_client`` is ``None``, a new dask client is created.
170
176
171
177
dask_client : dask.distributed.Client, optional
172
178
User-created dask client, can be used to start a dask cluster and then
@@ -182,7 +188,7 @@ def __init__(
182
188
* ``'y_optimization'`` : do not save the predictions for the
183
189
optimization/validation set, which would later on be used to build
184
190
an ensemble.
185
- * ``' model' `` : do not save any model files
191
+ * ``model`` : do not save any model files
186
192
187
193
smac_scenario_args : dict, optional (None)
188
194
Additional arguments inserted into the scenario of SMAC. See the
@@ -559,7 +565,7 @@ def leaderboard(
559
565
Gives an overview of all models trained during the search process along
560
566
with various statistics about their training.
561
567
562
- The availble statistics are:
568
+ The available statistics are:
563
569
564
570
**Simple**:
565
571
0 commit comments