Skip to content

Commit 953871d

Browse files
authored
remove num_class hyperparameter
num_class=1 gives error for binary classifcation task on sagemaker. Removing it runs the training job sucessfully.
1 parent f86a97f commit 953871d

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

introduction_to_applying_machine_learning/xgboost_direct_marketing/xgboost_direct_marketing_sagemaker.ipynb

-1
Original file line numberDiff line numberDiff line change
@@ -460,7 +460,6 @@
460460
" subsample=0.8,\n",
461461
" silent=0,\n",
462462
" objective='binary:logistic',\n",
463-
" num_class=1, \n",
464463
" num_round=100)\n",
465464
"\n",
466465
"xgb.fit({'train': s3_input_train, 'validation': s3_input_validation}) "

0 commit comments

Comments
 (0)