search

Snowflake-Labs / sfguide-intro-to-machine-learning-with-snowflake-ml-for-python

Apache License 2.0
54 stars 114 forks source link

XGBoost 1.7.6 vs XGBoost 2.0.3 #20

Open tomribaroff opened 4 months ago

tomribaroff commented 4 months ago

Using snowflake-ml-python==1.4.0 requires us to use XGBoost<2

My XGBoost model for a personal project is using the package version 2.0.3, and is running much faster than the 1.7.6 version (<2).

Are there any plans to update this, so that I can register my XGBoost model to the Model Registry using snowflake-ml-python?

Is there any work around for getting this model into the registry in the mean time? Or shall I retrain on 1.7.6 and deal with the slower run speeds for now?