treerite 0.1.0

Rust binding for treelite runtime (https://github.com/dmlc/treelite)
docs.rs failed to build treerite-0.1.0
Please check the build logs for more information.
See Builds for ideas on how to fix a failed build, or Metadata for how to configure docs.rs builds.
If you believe this is docs.rs' fault, open an issue.

TreeRite: TreeLite runtime in Rust CI

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