📜  如何在 MacOS 上安装 XGBoost 和 LightGBM?(1)

📅  最后修改于: 2023-12-03 14:52:26.400000             🧑  作者: Mango

如何在 MacOS 上安装 XGBoost 和 LightGBM?

XGBoost 和 LightGBM 都是常用的梯度提升算法库,它们可以用于分类、回归和排名等任务。在 MacOS 上安装 XGBoost 和 LightGBM 可以分为以下几个步骤。

安装依赖项

在安装 XGBoost 和 LightGBM 之前,需要安装一些依赖项。可以使用 Homebrew 来安装这些依赖项。

brew install cmake gcc
安装 XGBoost
从源代码安装
git clone --recursive https://github.com/dmlc/xgboost
cd xgboost
mkdir build
cd build
cmake .. -DUSE_OPENMP=1
make -j$(sysctl -n machdep.cpu.thread_count)

构建完成之后,可以在 python-package 目录下找到 Python 包,并将其安装到 Python 环境中:

cd python-package
python setup.py install
使用 pip 安装

也可以使用 pip 安装 XGBoost:

pip install xgboost
安装 LightGBM
从源代码安装
git clone --recursive https://github.com/Microsoft/LightGBM
cd LightGBM
mkdir build
cd build
cmake .. -DUSE_OPENMP=1
make -j$(sysctl -n machdep.cpu.thread_count)

构建完成之后,可以在 python-package 目录下找到 Python 包,并将其安装到 Python 环境中:

cd ../python-package
python setup.py install
使用 pip 安装

也可以使用 pip 安装 LightGBM:

pip install lightgbm

注意,使用 pip 安装 LightGBM 时需要安装 wheel 工具:

pip install wheel

以上是在 MacOS 上安装 XGBoost 和 LightGBM 的一些基本步骤,如果想要在自己的项目中使用这两个库,还需要根据具体需求进行相应的配置和调试。