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