https://github.com/vaaaaanquish/lightgbm-rs
LightGBM Rust binding
https://github.com/vaaaaanquish/lightgbm-rs
Last synced: 1 day ago
JSON representation
LightGBM Rust binding
- Host: GitHub
- URL: https://github.com/vaaaaanquish/lightgbm-rs
- Owner: vaaaaanquish
- License: mit
- Created: 2021-01-03T11:45:47.000Z (over 4 years ago)
- Default Branch: main
- Last Pushed: 2023-10-31T09:03:31.000Z (over 1 year ago)
- Last Synced: 2024-04-14T07:20:47.645Z (about 1 year ago)
- Language: Rust
- Size: 67.4 KB
- Stars: 67
- Watchers: 5
- Forks: 28
- Open Issues: 16
-
Metadata Files:
- Readme: README.md
- Funding: .github/FUNDING.yml
- License: LICENSE
Awesome Lists containing this project
README
# lightgbm-rs
LightGBM Rust binding
# Require
You need an environment that can build LightGBM.
```
# linux
apt install -y cmake libclang-dev libc++-dev gcc-multilib
# OS X
brew install cmake libomp
```
On Windows
1. Install CMake and VS Build Tools.
1. Install LLVM and set an environment variable `LIBCLANG_PATH` to PATH_TO_LLVM_BINARY (example: `C:\Program Files\LLVM\bin`)
Please see below for details.
- [LightGBM Installation-Guide](https://lightgbm.readthedocs.io/en/latest/Installation-Guide.html)
# Usage
Example LightGBM train.
```
extern crate serde_json;
use lightgbm::{Dataset, Booster};
use serde_json::json;
let data = vec![vec![1.0, 0.1, 0.2, 0.1],
vec![0.7, 0.4, 0.5, 0.1],
vec![0.9, 0.8, 0.5, 0.1],
vec![0.2, 0.2, 0.8, 0.7],
vec![0.1, 0.7, 1.0, 0.9]];
let label = vec![0.0, 0.0, 0.0, 1.0, 1.0];
let dataset = Dataset::from_mat(data, label).unwrap();
let params = json!{
{
"num_iterations": 3,
"objective": "binary",
"metric": "auc"
}
};
let bst = Booster::train(dataset, ¶ms).unwrap();
```
Please see the `./examples` for details.
|example|link|
|---|---|
|binary classification|[link](https://github.com/vaaaaanquish/lightgbm-rs/blob/main/examples/binary_classification/src/main.rs)|
|multiclass classification|[link](https://github.com/vaaaaanquish/lightgbm-rs/blob/main/examples/multiclass_classification/src/main.rs)|
|regression|[link](https://github.com/vaaaaanquish/lightgbm-rs/blob/main/examples/regression/src/main.rs)|
# Develop
```
git clone --recursive https://github.com/vaaaaanquish/lightgbm-rs
```
```
docker build -t lgbmrs .
docker run -it -v $PWD:/app lgbmrs bash
# cargo build
```
# Thanks
Much reference was made to implementation and documentation. Thanks.
- [microsoft/LightGBM](https://github.com/microsoft/LightGBM)
- [davechallis/rust-xgboost](https://github.com/davechallis/rust-xgboost)