[pyspark] Add validation for param 'early_stopping_rounds' and 'validation_indicator_col' (#8250)

Signed-off-by: Weichen Xu <weichen.xu@databricks.com>
This commit is contained in:
WeichenXu
2022-09-26 17:43:03 +08:00
committed by GitHub
parent 0cd11b893a
commit ff71c69adf
2 changed files with 15 additions and 0 deletions

View File

@@ -1145,3 +1145,8 @@ class XgboostLocalTest(SparkTestCase):
num_workers=4,
)
classifier.fit(data_trans)
def test_early_stop_param_validation(self):
classifier = SparkXGBClassifier(early_stopping_rounds=1)
with pytest.raises(ValueError, match="early_stopping_rounds"):
classifier.fit(self.cls_df_train)