cuML C++ API: ML::fil::detail::decision_forest_builder< decision_forest_t > Struct Template Reference

#include <decision_forest_builder.hpp>

Collaboration diagram for ML::fil::detail::decision_forest_builder< decision_forest_t >:

Collaboration graph

Public Types
using	node_type = typename decision_forest_t::node_type

Public Member Functions
template<typename iter_t >
void	add_categorical_node (iter_t vec_begin, iter_t vec_end, std::optional< int > tl_node_id=std::nullopt, std::size_t depth=std::size_t{1}, bool default_to_distant_child=false, typename node_type::metadata_storage_type feature=typename node_type::metadata_storage_type{}, typename node_type::offset_type offset=typename node_type::offset_type{})

template<typename iter_t >
void	add_leaf_vector_node (iter_t vec_begin, iter_t vec_end, std::optional< int > tl_node_id=std::nullopt, std::size_t depth=std::size_t{1})

template<typename value_t >
void	add_node (value_t val, std::optional< int > tl_node_id=std::nullopt, std::size_t depth=std::size_t{1}, bool is_leaf_node=true, bool default_to_distant_child=false, bool is_categorical_node=false, typename node_type::metadata_storage_type feature=typename node_type::metadata_storage_type{}, typename node_type::offset_type offset=typename node_type::offset_type{}, bool is_inclusive=false)

void	set_element_postproc (element_op val)

void	set_row_postproc (row_op val)

void	set_average_factor (double val)

void	set_bias (double val)

void	set_postproc_constant (double val)

void	set_output_size (index_type val)

	decision_forest_builder (index_type max_num_categories=index_type{}, index_type align_bytes=index_type{})

auto	get_decision_forest (index_type num_feature, index_type num_class, raft_proto::device_type mem_type=raft_proto::device_type::cpu, int device=0, raft_proto::cuda_stream stream=raft_proto::cuda_stream{})

Member Typedef Documentation

◆ node_type

template<typename decision_forest_t >

using ML::fil::detail::decision_forest_builder< decision_forest_t >::node_type = typename decision_forest_t::node_type

Constructor & Destructor Documentation

◆ decision_forest_builder()

template<typename decision_forest_t >

ML::fil::detail::decision_forest_builder< decision_forest_t >::decision_forest_builder	(	index_type	max_num_categories = `index_type{}`,
		index_type	align_bytes = `index_type{}`
	)

inline

Member Function Documentation

◆ add_categorical_node()

template<typename decision_forest_t >

template<typename iter_t >

void ML::fil::detail::decision_forest_builder< decision_forest_t >::add_categorical_node	(	iter_t	vec_begin,
		iter_t	vec_end,
		std::optional< int >	tl_node_id = `std::nullopt`,
		std::size_t	depth = `std::size_t{1}`,
		bool	default_to_distant_child = `false`,
		typename node_type::metadata_storage_type	feature = `typename node_type::metadata_storage_type{}`,
		typename node_type::offset_type	offset = `typename node_type::offset_type{}`
	)

inline

◆ add_leaf_vector_node()

template<typename decision_forest_t >

template<typename iter_t >

void ML::fil::detail::decision_forest_builder< decision_forest_t >::add_leaf_vector_node	(	iter_t	vec_begin,
		iter_t	vec_end,
		std::optional< int >	tl_node_id = `std::nullopt`,
		std::size_t	depth = `std::size_t{1}`
	)

inline

◆ add_node()

template<typename decision_forest_t >

template<typename value_t >

void ML::fil::detail::decision_forest_builder< decision_forest_t >::add_node	(	value_t	val,
		std::optional< int >	tl_node_id = `std::nullopt`,
		std::size_t	depth = `std::size_t{1}`,
		bool	is_leaf_node = `true`,
		bool	default_to_distant_child = `false`,
		bool	is_categorical_node = `false`,
		typename node_type::metadata_storage_type	feature = `typename node_type::metadata_storage_type{}`,
		typename node_type::offset_type	offset = `typename node_type::offset_type{}`,
		bool	is_inclusive = `false`
	)

inline

◆ get_decision_forest()

template<typename decision_forest_t >

auto ML::fil::detail::decision_forest_builder< decision_forest_t >::get_decision_forest	(	index_type	num_feature,
		index_type	num_class,
		raft_proto::device_type	mem_type = `raft_proto::device_type::cpu`,
		int	device = `0`,
		raft_proto::cuda_stream	stream = `raft_proto::cuda_stream{}`
	)

inline

◆ set_average_factor()

template<typename decision_forest_t >

void ML::fil::detail::decision_forest_builder< decision_forest_t >::set_average_factor ( double val )

inline

◆ set_bias()

template<typename decision_forest_t >

void ML::fil::detail::decision_forest_builder< decision_forest_t >::set_bias ( double val )

inline

◆ set_element_postproc()

template<typename decision_forest_t >

void ML::fil::detail::decision_forest_builder< decision_forest_t >::set_element_postproc ( element_op val )

inline

◆ set_output_size()

template<typename decision_forest_t >

void ML::fil::detail::decision_forest_builder< decision_forest_t >::set_output_size ( index_type val )

inline

◆ set_postproc_constant()

template<typename decision_forest_t >

void ML::fil::detail::decision_forest_builder< decision_forest_t >::set_postproc_constant ( double val )

inline

◆ set_row_postproc()

template<typename decision_forest_t >

void ML::fil::detail::decision_forest_builder< decision_forest_t >::set_row_postproc ( row_op val )

inline

The documentation for this struct was generated from the following file:

include/cuml/fil/detail/decision_forest_builder.hpp