StochTree 0.0.1
Loading...
Searching...
No Matches
StochTree::Tree Member List

This is the complete list of members for StochTree::Tree, including all inherited members.

AllocNode()StochTree::Tree
CategoryList(std::int32_t nid) constStochTree::Treeinline
CloneFromTree(Tree *tree)StochTree::Tree
CollapseToLeaf(std::int32_t nid, double value)StochTree::Treeinline
CollapseToLeaf(std::int32_t nid, std::vector< double > value_vector)StochTree::Treeinline
DefaultChild(std::int32_t nid) constStochTree::Treeinline
DeleteNode(std::int32_t nid)StochTree::Tree
ExpandNode(std::int32_t nid, int split_index, double split_value, double left_value, double right_value)StochTree::Tree
ExpandNode(std::int32_t nid, int split_index, std::vector< std::uint32_t > const &categorical_indices, double left_value, double right_value)StochTree::Tree
ExpandNode(std::int32_t nid, int split_index, double split_value, std::vector< double > left_value_vector, std::vector< double > right_value_vector)StochTree::Tree
ExpandNode(std::int32_t nid, int split_index, std::vector< std::uint32_t > const &categorical_indices, std::vector< double > left_value_vector, std::vector< double > right_value_vector)StochTree::Tree
ExpandNode(std::int32_t nid, int split_index, TreeSplit &split, double left_value, double right_value)StochTree::Tree
ExpandNode(std::int32_t nid, int split_index, TreeSplit &split, std::vector< double > left_value_vector, std::vector< double > right_value_vector)StochTree::Tree
from_json(const json &tree_json)StochTree::Tree
GetDepth(std::int32_t nid) constStochTree::Treeinline
GetInternalNodes() constStochTree::Treeinline
GetLeafParents() constStochTree::Treeinline
GetLeaves() constStochTree::Treeinline
GetNodes()StochTree::Treeinline
HasCategoricalSplit() constStochTree::Treeinline
HasLeafVector(std::int32_t nid) constStochTree::Treeinline
HasVectorOutput() constStochTree::Treeinline
Init(int output_dimension=1, bool is_log_scale=false)StochTree::Tree
IsCategoricalSplitNode(std::int32_t nid) constStochTree::Treeinline
IsDeleted(std::int32_t nid) constStochTree::Treeinline
IsLeaf(std::int32_t nid) constStochTree::Treeinline
IsLogScale() constStochTree::Treeinline
IsNumericSplitNode(std::int32_t nid) constStochTree::Treeinline
IsRoot()StochTree::Treeinline
IsRoot(std::int32_t nid) constStochTree::Treeinline
LeafValue(std::int32_t nid) constStochTree::Treeinline
LeafValue(std::int32_t nid, std::int32_t dim_id) constStochTree::Treeinline
LeafVector(std::int32_t nid) constStochTree::Treeinline
LeftChild(std::int32_t nid) constStochTree::Treeinline
MaxLeafDepth() constStochTree::Treeinline
NodeType(std::int32_t nid) constStochTree::Treeinline
NumDeletedNodes() const noexceptStochTree::Treeinline
NumNodes() const noexceptStochTree::Treeinline
NumValidNodes() const noexceptStochTree::Treeinline
OutputDimension() constStochTree::Treeinline
Parent(std::int32_t nid) constStochTree::Treeinline
PredictLeafIndexInplace(ForestDataset *dataset, std::vector< int32_t > &output, int32_t offset, int32_t max_leaf)StochTree::Tree
PredictLeafIndexInplace(Eigen::MatrixXd &covariates, std::vector< int32_t > &output, int32_t offset, int32_t max_leaf)StochTree::Tree
PredictLeafIndexInplace(Eigen::Map< Eigen::Matrix< double, Eigen::Dynamic, Eigen::Dynamic, Eigen::ColMajor > > &covariates, std::vector< int32_t > &output, int32_t offset, int32_t max_leaf)StochTree::Tree
Reset()StochTree::Tree
RightChild(std::int32_t nid) constStochTree::Treeinline
SetCategoricalSplit(std::int32_t nid, std::int32_t split_index, std::vector< std::uint32_t > const &category_list)StochTree::Tree
SetChildren(std::int32_t nid, std::int32_t left_child, std::int32_t right_child)StochTree::Treeinline
SetLeaf(std::int32_t nid, double value)StochTree::Tree
SetLeafVector(std::int32_t nid, std::vector< double > const &leaf_vector)StochTree::Tree
SetLeftChild(std::int32_t nid, std::int32_t left_child)StochTree::Treeinline
SetNumericSplit(std::int32_t nid, std::int32_t split_index, double threshold)StochTree::Tree
SetParent(std::int32_t child_node, std::int32_t parent_node)StochTree::Treeinline
SetParents(std::int32_t nid, std::int32_t left_child, std::int32_t right_child)StochTree::Treeinline
SetRightChild(std::int32_t nid, std::int32_t right_child)StochTree::Treeinline
SplitIndex(std::int32_t nid) constStochTree::Treeinline
SumSquaredLeafValues() constStochTree::Treeinline
SumSquaredNodeValues(std::int32_t nid) constStochTree::Treeinline
Threshold(std::int32_t nid) constStochTree::Treeinline
to_json()StochTree::Tree
WalkTree(Func func) constStochTree::Treeinline