XGBoost和CatBoost是Kaggle等算法比赛里常用的杀器,算法性能比一般的回归和分类算法都要好。简而言之,当你在解决一些传统的机器学习回归或分类问题的时候,想不到用什么算法的时候,可以试试这两个算法,效果还不错,至少快速完成一个baseline是没有问题的。
1、安装xgboost
pip3 install xgboost -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
Looking in indexes: http://pypi.douban.com/simple/
Collecting xgboost
Downloading http://pypi.doubanio.com/packages/92/26/d8643f1785d42cd37d7b0a7c20ca38bc452f4866a6931b61b81af9b5493c/xgboost-1.6.2-py3-none-macosx_10_15_x86_64.macosx_11_0_x86_64.macosx_12_0_x86_64.whl (1.7 MB)
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 1.7/1.7 MB 4.1 MB/s eta 0:00:00
Requirement already satisfied: scipy in /Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages (from xgboost) (1.7.3)
Requirement already satisfied: numpy in /Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages (from xgboost) (1.21.6)
Installing collected packages: xgboost
Successfully installed xgboost-1.6.2
3、安装catboost
pip3 install catboost -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
4、查看xgboost和catboost版本
>>> import xgboost
>>> xgboost.__version__
'1.6.2'
>>> import catboost
>>> catboost.__version__
'1.0.6'