categorical_storage_type typedef | ML::experimental::fil::decision_forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t > | |
decision_forest() | ML::experimental::fil::decision_forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t > | inline |
decision_forest(raft_proto::buffer< node_type > &&nodes, raft_proto::buffer< index_type > &&root_node_indexes, raft_proto::buffer< index_type > &&node_id_mapping, index_type num_features, index_type num_outputs=index_type{2}, bool has_categorical_nodes=false, std::optional< raft_proto::buffer< io_type >> &&vector_output=std::nullopt, std::optional< raft_proto::buffer< typename node_type::index_type >> &&categorical_storage=std::nullopt, index_type leaf_size=index_type{1}, row_op row_postproc=row_op::disable, element_op elem_postproc=element_op::disable, io_type average_factor=io_type{1}, io_type bias=io_type{0}, io_type postproc_constant=io_type{1}) | ML::experimental::fil::decision_forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t > | inline |
device_index() | ML::experimental::fil::decision_forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t > | inline |
elem_postprocessing() const | ML::experimental::fil::decision_forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t > | inline |
forest_type typedef | ML::experimental::fil::decision_forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t > | |
has_vector_leaves() const | ML::experimental::fil::decision_forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t > | inline |
io_type typedef | ML::experimental::fil::decision_forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t > | |
layout | ML::experimental::fil::decision_forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t > | static |
memory_type() | ML::experimental::fil::decision_forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t > | inline |
node_type typedef | ML::experimental::fil::decision_forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t > | |
num_features() const | ML::experimental::fil::decision_forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t > | inline |
num_outputs(infer_kind inference_kind=infer_kind::default_kind) const | ML::experimental::fil::decision_forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t > | inline |
num_trees() const | ML::experimental::fil::decision_forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t > | inline |
postprocessor_type typedef | ML::experimental::fil::decision_forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t > | |
predict(raft_proto::buffer< typename forest_type::io_type > &output, raft_proto::buffer< typename forest_type::io_type > const &input, raft_proto::cuda_stream stream=raft_proto::cuda_stream{}, infer_kind predict_type=infer_kind::default_kind, std::optional< index_type > specified_rows_per_block_iter=std::nullopt) | ML::experimental::fil::decision_forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t > | inline |
row_postprocessing() const | ML::experimental::fil::decision_forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t > | inline |
set_row_postprocessing(row_op val) | ML::experimental::fil::decision_forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t > | inline |
threshold_type typedef | ML::experimental::fil::decision_forest< layout_v, threshold_t, index_t, metadata_storage_t, offset_t > | |