ML::fil::forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t > Struct Template Reference

#include <forest.hpp>

Collaboration diagram for ML::fil::forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t >:

Public Types
using	node_type = node< layout_v, threshold_t, index_t, metadata_storage_t, offset_t >
using	io_type = threshold_t
template<typename vector_output_t >
using	raw_output_type = std::conditional_t<!std::is_same_v< vector_output_t, std::nullptr_t >, std::remove_pointer_t< vector_output_t >, typename node_type::threshold_type >

Public Member Functions
HOST DEVICE	forest (node_type *forest_nodes, index_type *forest_root_indexes, index_type *node_id_mapping, index_type num_trees, index_type num_outputs)
HOST DEVICE auto *	get_tree_root (index_type tree_index) const
HOST DEVICE const auto *	get_node_id_mapping () const
HOST DEVICE auto	tree_count () const
HOST DEVICE auto	num_outputs () const

Member Typedef Documentation

io_type

template<tree_layout layout_v, typename threshold_t , typename index_t , typename metadata_storage_t , typename offset_t >

using ML::fil::forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t >::io_type = threshold_t

node_type

template<tree_layout layout_v, typename threshold_t , typename index_t , typename metadata_storage_t , typename offset_t >

using ML::fil::forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t >::node_type = node<layout_v, threshold_t, index_t, metadata_storage_t, offset_t>

raw_output_type

template<tree_layout layout_v, typename threshold_t , typename index_t , typename metadata_storage_t , typename offset_t >

template<typename vector_output_t >

using ML::fil::forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t >::raw_output_type = std::conditional_t<!std::is_same_v<vector_output_t, std::nullptr_t>, std::remove_pointer_t<vector_output_t>, typename node_type::threshold_type>

Constructor & Destructor Documentation

forest()

template<tree_layout layout_v, typename threshold_t , typename index_t , typename metadata_storage_t , typename offset_t >

HOST DEVICE ML::fil::forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t >::forest ( node_type *  forest_nodes, index_type *  forest_root_indexes, index_type *  node_id_mapping, index_type  num_trees, index_type  num_outputs  )

inline

Member Function Documentation

get_node_id_mapping()

template<tree_layout layout_v, typename threshold_t , typename index_t , typename metadata_storage_t , typename offset_t >

HOST DEVICE const auto* ML::fil::forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t >::get_node_id_mapping ( ) const

inline

get_tree_root()

template<tree_layout layout_v, typename threshold_t , typename index_t , typename metadata_storage_t , typename offset_t >

HOST DEVICE auto* ML::fil::forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t >::get_tree_root ( index_type  tree_index ) const

inline

num_outputs()

template<tree_layout layout_v, typename threshold_t , typename index_t , typename metadata_storage_t , typename offset_t >

HOST DEVICE auto ML::fil::forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t >::num_outputs ( ) const

inline

tree_count()

template<tree_layout layout_v, typename threshold_t , typename index_t , typename metadata_storage_t , typename offset_t >

HOST DEVICE auto ML::fil::forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t >::tree_count ( ) const

inline

---

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

• include/cuml/fil/detail/forest.hpp