Treelite
|
Branch annotation tools. More...
#include <treelite/logging.h>
#include <treelite/annotator.h>
#include <treelite/math.h>
#include <rapidjson/istreamwrapper.h>
#include <rapidjson/ostreamwrapper.h>
#include <rapidjson/writer.h>
#include <rapidjson/document.h>
#include <limits>
#include <thread>
#include <cstdint>
#include "threading_utils/parallel_for.h"
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< uint64_t >> *out_counts) |