1 unstable release
0.1.0 | Jan 13, 2021 |
---|
#1785 in Algorithms
5MB
12K
SLoC
Contains (ELF lib, 29KB) examples/iris.so
TreeRite: TreeLite runtime in Rust
This binding currently works for treelite 1.0.0rc1
Usage
- Make sure your machine has cmake, libstdc++ and libgomp installed.
- Add this to your project's
Cargo.toml
.treerite = { git = "https://github.com/dovahcrow/treerite" }
By default, the treerite library is static linked to your binary. If you'd like to use the dynamic lib,
set the dynamic feature of treerite
.
Documentation
There's no documentation available yet. But you can take a look at the example folder and the tests folder.
The usage should be quite straight forward: you first load the Predictor
from a shared library of the model.
Then, load the data from Vec
or ndarray::Array2
into DMatrix
. Finally, you do prediction using
Predictor::predict_batch(dmatrix)
.
TODO
- TreeliteDMatrixCreateFromCSR
- TreeliteDMatrixCreateFromFile