📜  如何在 Linux 上安装 Lightgbm?(1)

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

如何在 Linux 上安装 Lightgbm?

LightGBM 是一个开源的分布式梯度提升框架,具有高效、低内存占用和高精度等特点,广泛应用于机器学习、推荐系统等领域。本文将介绍如何在 Linux 上安装 LightGBM。

安装依赖

首先,需要安装一些必要的依赖:

sudo apt install cmake g++
sudo apt install libboost-dev libboost-system-dev libboost-filesystem-dev
下载和编译源代码

在安装完依赖后,可以从 GitHub 上下载 LightGBM 的源代码:

git clone --recursive https://github.com/microsoft/LightGBM

这里需要注意,LightGBM 的源代码采用了子模块的方式管理,加上 --recursive 参数可以将子模块也一起下载。

下载完成后,进入源代码目录,执行以下命令进行编译:

cd LightGBM
mkdir build
cd build
cmake ..
make -j4

这里的 -j4 表示使用 4 个线程进行编译,可以根据自己的 CPU 核数进行调整。编译完成后,可以运行以下命令进行测试:

cd ../examples/cpp_classification
mkdir build
cd build
cmake ..
make -j4
./lightgbm

如果测试运行成功,就可以将编译好的库文件和头文件拷贝到系统库目录和头文件目录中:

sudo cp ../lib_lightgbm.so /usr/local/lib
sudo cp ../include/* /usr/local/include
安装 Python 包

除了 C++ 的接口外,LightGBM 也提供了 Python 包以供调用。可以使用 pip 命令安装:

pip install lightgbm

安装完成后,可以在 Python 中导入 LightGBM 库并进行使用:

import lightgbm as lgb

至此,LightGBM 在 Linux 上的安装就完成了。

总结

本文介绍了如何在 Linux 上安装 LightGBM,主要包括以下几个步骤:

  1. 安装依赖
  2. 下载和编译源代码
  3. 将库文件和头文件拷贝到系统目录中
  4. 安装 Python 包

希望本文对大家有所帮助。