TreeRite: TreeLite运行时在Rust中的绑定

此绑定目前适用于treelite 1.0.0rc1

使用方法

• 确保您的机器已安装cmake、libstdc++和libgomp。
• 将以下内容添加到项目的Cargo.toml中。

  treerite = { git = "https://github.com/dovahcrow/treerite" }
  

默认情况下，treerite库会静态链接到您的二进制文件。如果您想使用动态库，请设置treerite的动态功能。

文档

目前尚无文档。但您可以查看示例文件夹和测试文件夹。使用方法应该相当简单：首先从模型的共享库中加载Predictor。然后，将数据从Vec或ndarray::Array2加载到DMatrix中。最后，使用Predictor::predict_batch(dmatrix)进行预测。

待办事项

• TreeliteDMatrixCreateFromCSR
• TreeliteDMatrixCreateFromFile