https://github.com/sh1ng/arboretum
Gradient Boosting powered by GPU(NVIDIA CUDA)
https://github.com/sh1ng/arboretum
arboretum cuda gpu gradient-boosting gradient-boosting-machine machine-learning python
Last synced: 8 months ago
JSON representation
Gradient Boosting powered by GPU(NVIDIA CUDA)
- Host: GitHub
- URL: https://github.com/sh1ng/arboretum
- Owner: sh1ng
- License: other
- Created: 2016-01-28T19:40:35.000Z (over 10 years ago)
- Default Branch: master
- Last Pushed: 2020-04-07T09:50:42.000Z (about 6 years ago)
- Last Synced: 2025-03-28T21:23:54.183Z (about 1 year ago)
- Topics: arboretum, cuda, gpu, gradient-boosting, gradient-boosting-machine, machine-learning, python
- Language: Cuda
- Homepage:
- Size: 656 KB
- Stars: 64
- Watchers: 3
- Forks: 14
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# arboretum - Gradient Boosting implementation with focus on overcoming GRAM size limit
## Installation wheel package
```
pip install arboretum
```
[Benchmark results](https://bit.ly/2VcUwEV)
## Dependencies
* Python 2.7 or Python 3
* Cuda toolkit 7+
* Cuda cub https://github.com/NVlabs/cub as a submodule
* JSON for Modern C++ https://github.com/nlohmann/json as a submodule
## Installation from source
* git clone --recursive https://github.com/sh1ng/arboretum.git
* $ mkdir build && cd build && cmake .. && make -j && cd .. && make wheel
* $ sudo python -m pip install python-package/dist/arboretum*.whl