Treelite
|
Branch annotation tools. More...
#include <treelite/annotator.h>
#include <treelite/math.h>
#include <treelite/omp.h>
#include <dmlc/json.h>
#include <limits>
#include <cstdint>
Go to the source code of this file.
Functions | |
template<typename ThresholdType , typename LeafOutputType > | |
void | AnnotateImpl (const treelite::ModelImpl< ThresholdType, LeafOutputType > &model, const treelite::DMatrix *dmat, int nthread, int verbose, std::vector< std::vector< size_t >> *out_counts) |