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: 3 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 (about 9 years ago)
• Default Branch: master
• Last Pushed: 2020-04-07T09:50:42.000Z (almost 5 years ago)
• Last Synced: 2024-11-08T12:04:28.438Z (3 months ago)
• Topics: arboretum, cuda, gpu, gradient-boosting, gradient-boosting-machine, machine-learning, python
• Language: Cuda
• Homepage:
• Size: 656 KB
• Stars: 64
• Watchers: 4
• 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