tree.h File Reference

model structure for tree ensemble More...

#include <treelite/contiguous_array.h>
#include <treelite/enum/operator.h>
#include <treelite/enum/task_type.h>
#include <treelite/enum/tree_node_type.h>
#include <treelite/enum/typeinfo.h>
#include <treelite/logging.h>
#include <treelite/pybuffer_frame.h>
#include <treelite/version.h>
#include <algorithm>
#include <cstddef>
#include <cstdint>
#include <cstdio>
#include <iostream>
#include <limits>
#include <map>
#include <memory>
#include <sstream>
#include <string>
#include <type_traits>
#include <utility>
#include <variant>
#include <vector>
#include <treelite/detail/tree.h>

Include dependency graph for tree.h:

Go to the source code of this file.

Classes
struct	treelite::Version
class	treelite::Tree< ThresholdType, LeafOutputType >
	in-memory representation of a decision tree More...
class	treelite::ModelPreset< ThresholdT, LeafOutputT >
	Typed portion of the model class. More...
class	treelite::Model
	Model class for tree ensemble model. More...

Namespaces
	treelite
	treelite::detail
	treelite::detail::serializer
	treelite::detail::field_accessor

Macros
#define	TREELITE_DLL_EXPORT

Typedefs
using	treelite::ModelPresetVariant = std::variant< ModelPreset< float, float >, ModelPreset< double, double > >

Functions
template<typename ThresholdType , typename LeafOutputType >
PyBufferFrame	treelite::detail::field_accessor::GetTreeFieldImpl (ModelPreset< ThresholdType, LeafOutputType > &, std::uint64_t, std::string const &)
template<typename ThresholdType , typename LeafOutputType >
void	treelite::detail::field_accessor::SetTreeFieldImpl (ModelPreset< ThresholdType, LeafOutputType > &, std::uint64_t, std::string const &, PyBufferFrame)
template<int variant_index>
ModelPresetVariant	treelite::SetModelPresetVariant (int target_variant_index)
std::unique_ptr< Model >	treelite::ConcatenateModelObjects (std::vector< Model const * > const &objs)
	Concatenate multiple model objects into a single model object by copying all member trees into the destination model object. More...

Detailed Description

model structure for tree ensemble

Copyright (c) 2017-2023 by Contributors

Author

Hyunsu Cho

Macro Definition Documentation

TREELITE_DLL_EXPORT

#define TREELITE_DLL_EXPORT