7 #ifndef TREELITE_ANNOTATOR_H_ 8 #define TREELITE_ANNOTATOR_H_ 28 int nthread,
int verbose);
33 void Load(dmlc::Stream* fi);
38 void Save(dmlc::Stream* fo)
const;
52 inline std::vector<std::vector<size_t>>
Get()
const {
57 std::vector<std::vector<size_t>> counts;
62 #endif // TREELITE_ANNOTATOR_H_
thin wrapper for tree ensemble model
Input data structure of treelite.
void Annotate(const Model &model, const DMatrix *dmat, int nthread, int verbose)
annotate branches in a given model using frequency patterns in the training data. The annotation can ...
std::vector< std::vector< size_t > > Get() const
fetch branch annotation. Usage example:
void Save(dmlc::Stream *fo) const
save branch annotation to a JSON file
a simple data matrix in CSR (Compressed Sparse Row) storage
void Load(dmlc::Stream *fi)
load branch annotation from a JSON file