Treelite
Functions
annotator.cc File Reference

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"
Include dependency graph for annotator.cc:

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)
 

Detailed Description

Branch annotation tools.

Copyright (c) 2017-2021 by Contributors

Author
Hyunsu Cho

Definition in file annotator.cc.