LightGBM安装 - Shell/Bash

LightGBM是一个高效的梯度提升框架，具有高效、低内存占用、快速训练速度等优势，适用于大型数据集和低延迟应用场景。本文将介绍LightGBM的安装方法。

安装前准备

在安装LightGBM之前，需要在您的系统上安装cmake和gcc/g++。您可在终端中输入以下命令来检测它们是否已安装：

# 检测cmake安装情况
cmake --version

# 检测gcc安装情况
gcc --version

# 检测g++安装情况
g++ --version

如果您的系统中没有安装上述依赖，您需要按以下方式安装：

• 在Ubuntu上：

# 安装cmake
sudo apt-get install cmake

# 安装gcc/g++
sudo apt-get install gcc g++

• 在CentOS/RHEL上：

# 安装cmake
sudo yum install cmake

# 安装gcc/g++
sudo yum install gcc-c++ gcc

• 在macOS上：

# 安装cmake
brew install cmake

# 安装gcc/g++
brew install gcc

安装LightGBM

您可以按以下步骤来下载并编译安装LightGBM：

1. 克隆LightGBM源代码仓库：

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

2. 进入代码目录并创建build文件夹：

cd LightGBM
mkdir build
cd build

3. 使用cmake生成Makefile并编译：

cmake ..
make -j4

4. （可选）运行单元测试：

cd LightGBM/tests/cpp_test
./run_tests.sh

5. （可选）安装LightGBM：

sudo make install

在Python中使用LightGBM

除了用C++编写LightGBM模型，您还可以使用Python的LightGBM包。只需在Python中安装LightGBM包即可。

pip install lightgbm

您还需要安装numpy、scipy和scikit-learn等机器学习包。您可以通过以下命令来安装它们：

pip install numpy scipy scikit-learn

结论

LightGBM是一款高效、快速的机器学习框架，有助于加速数据科学的流程。通过本文的介绍，您了解了如何在Linux和macOS中安装和使用LightGBM。现在，您可以开始使用它来进行数据建模和预测。