treelite.hpp File Reference

#include <cuml/experimental/forest/exceptions.hpp>
#include <cuml/experimental/forest/traversal/traversal_forest.hpp>
#include <cuml/experimental/forest/traversal/traversal_node.hpp>
#include <treelite/tree.h>
#include <algorithm>
#include <numeric>
#include <vector>

Include dependency graph for treelite.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	ML::experimental::forest::treelite_traversal_node< tl_threshold_t, tl_output_t >
struct	ML::experimental::forest::treelite_traversal_forest< tl_threshold_t, tl_output_t >

Namespaces
	ML
	ML::experimental
	ML::experimental::forest

Typedefs
using	ML::experimental::forest::TREELITE_NODE_ID_T = int

Functions
template<typename lambda_t >
void	ML::experimental::forest::tree_for_each (treelite::Model const &tl_model, lambda_t &&lambda)
template<typename iter_t , typename lambda_t >
void	ML::experimental::forest::tree_transform (treelite::Model const &tl_model, iter_t out_iter, lambda_t &&lambda)
template<typename T , typename lambda_t >
auto	ML::experimental::forest::tree_accumulate (treelite::Model const &tl_model, T init, lambda_t &&lambda)
template<forest_order order, typename lambda_t >
void	ML::experimental::forest::node_for_each (treelite::Model const &tl_model, lambda_t &&lambda)
template<forest_order order, typename iter_t , typename lambda_t >
void	ML::experimental::forest::node_transform (treelite::Model const &tl_model, iter_t output_iter, lambda_t &&lambda)
template<forest_order order, typename T , typename lambda_t >
auto	ML::experimental::forest::node_accumulate (treelite::Model const &tl_model, T init, lambda_t &&lambda)