7 #ifndef TREELITE_ANNOTATOR_H_ 8 #define TREELITE_ANNOTATOR_H_ 36 void Load(std::istream& fi);
41 void Save(std::ostream& fo)
const;
55 inline std::vector<std::vector<uint64_t>>
Get()
const {
60 std::vector<std::vector<uint64_t>> counts_;
65 #endif // TREELITE_ANNOTATOR_H_
void Save(std::ostream &fo) const
save branch annotation to a JSON file
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 ...
model structure for tree ensemble
std::vector< std::vector< uint64_t > > Get() const
fetch branch annotation. Usage example:
thin wrapper for tree ensemble model
void Load(std::istream &fi)
load branch annotation from a JSON file