fil.h File Reference

#include <cuml/ensemble/treelite_defs.hpp>
#include <stddef.h>
#include <variant>

Include dependency graph for fil.h:

Go to the source code of this file.

Classes
struct	ML::fil::treelite_params_t

Namespaces
	raft
	ML
	ML::fil

Typedefs
template<typename real_t >
using	ML::fil::forest_t = forest< real_t > *
using	ML::fil::forest32_t = forest< float > *
using	ML::fil::forest64_t = forest< double > *
using	ML::fil::forest_variant = std::variant< forest_t< float >, forest_t< double > >

Enumerations
enum	ML::fil::algo_t { ML::fil::ALGO_AUTO , ML::fil::NAIVE , ML::fil::TREE_REORG , ML::fil::BATCH_TREE_REORG }
enum	ML::fil::storage_type_t { ML::fil::AUTO , ML::fil::DENSE , ML::fil::SPARSE , ML::fil::SPARSE8 }
enum	ML::fil::precision_t { ML::fil::PRECISION_NATIVE , ML::fil::PRECISION_FLOAT32 , ML::fil::PRECISION_FLOAT64 }

Functions
void	ML::fil::from_treelite (const raft::handle_t &handle, forest_variant *pforest, TreeliteModelHandle model, const treelite_params_t *tl_params)
template<typename real_t >
void	ML::fil::free (const raft::handle_t &h, forest_t< real_t > f)
template<typename real_t >
void	ML::fil::predict (const raft::handle_t &h, forest_t< real_t > f, real_t *preds, const real_t *data, size_t num_rows, bool predict_proba=false)

Variables
constexpr int	ML::fil::MAX_N_ITEMS = 4

Detailed Description

Interface to the forest inference library.