8000 GitHub - sh1ng/arboretum: Gradient Boosting powered by GPU(NVIDIA CUDA)
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

sh1ng/arboretum

Repository files navigation

arboretum - Gradient Boosting implementation with focus on overcoming GRAM size limit

Installation wheel package

pip install arboretum

Benchmark results

Dependencies

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

About

Gradient Boosting powered by GPU(NVIDIA CUDA)

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Contributors 3

  •  
  •  
  •  
0