search

snowflakedb / snowflake-ml-python

Apache License 2.0
42 stars 12 forks source link

SNOW-1679397: Missing eval_set in XGBoost and in other estimators #117

Open andreidore opened 1 month ago

andreidore commented 1 month ago

Hi

In the XGBoost documentation (https://docs.snowflake.com/en/developer-guide/snowpark-ml/reference/latest/api/modeling/snowflake.ml.modeling.xgboost.XGBClassifier#snowflake.ml.modeling.xgboost.XGBClassifier), you specified that it is possible to pass an eval_set in the fit method to enable setting early_stopping_rounds. However, the fit method only allows a DataFrame object (the training DataFrame).

How is it possible to pass the eval_set?

Thanks

sfc-gh-xjiang commented 3 weeks ago

Hello,

Thank you for bringing this to our attention. Currently, the eval_set parameter is not supported in the fit method of the XGBClassifier in the Snowpark ML modeling library. We apologize for any confusion the documentation may have caused.

We appreciate your feedback and will work on updating the documentation to reflect the current capabilities of the library. Supporting eval_set and early stopping is an important feature, and we will consider it for future releases.

Please let us know if you have any other questions or need further assistance.

Best regards, Xinyi