Skip to content

Commit 1f31d3b

Browse files
authored
Merge pull request #106999 from Aniththa/patch-42
Update how-to-configure-auto-train.md
2 parents 8123bc6 + 0af108a commit 1f31d3b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

articles/machine-learning/how-to-configure-auto-train.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ Some examples include:
149149
task='classification',
150150
primary_metric='AUC_weighted',
151151
experiment_timeout_minutes=30,
152-
blacklist_models='XGBoostClassifier',
152+
blacklist_models=['XGBoostClassifier'],
153153
training_data=train_data,
154154
label_column_name=label,
155155
n_cross_validations=2)
@@ -160,7 +160,7 @@ Some examples include:
160160
automl_regressor = AutoMLConfig(
161161
task='regression',
162162
experiment_timeout_minutes=60,
163-
whitelist_models='kNN regressor'
163+
whitelist_models=['kNN regressor'],
164164
primary_metric='r2_score',
165165
training_data=train_data,
166166
label_column_name=label,

0 commit comments

Comments
 (0)