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

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

CloneFromExistingTree(int i, Tree *tree)StochTree::TreeEnsembleinline
from_json(const json &ensemble_json)StochTree::TreeEnsembleinline
GetMaxLeafIndex()StochTree::TreeEnsembleinline
GetTree(int i)StochTree::TreeEnsembleinline
PredictLeafIndices(ForestDataset *dataset)StochTree::TreeEnsembleinline
PredictLeafIndicesInplace(ForestDataset *dataset, std::vector< int32_t > &output, int num_trees, data_size_t n)StochTree::TreeEnsembleinline
PredictLeafIndicesInplace(Eigen::Map< Eigen::Matrix< double, Eigen::Dynamic, Eigen::Dynamic, Eigen::ColMajor > > &covariates, std::vector< int32_t > &output, int num_trees, data_size_t n)StochTree::TreeEnsembleinline
PredictLeafIndicesInplace(Eigen::Map< Eigen::Matrix< double, Eigen::Dynamic, Eigen::Dynamic, Eigen::ColMajor > > &covariates, Eigen::Map< Eigen::Matrix< int, Eigen::Dynamic, Eigen::Dynamic, Eigen::ColMajor > > &output, int column_ind, int num_trees, data_size_t n)StochTree::TreeEnsembleinline
PredictLeafIndicesInplace(Eigen::MatrixXd &covariates, std::vector< int32_t > &output, int num_trees, data_size_t n)StochTree::TreeEnsembleinline
ReconstituteFromForest(TreeEnsemble &ensemble)StochTree::TreeEnsembleinline
ResetInitTree(int i)StochTree::TreeEnsembleinline
ResetRoot()StochTree::TreeEnsembleinline
ResetTree(int i)StochTree::TreeEnsembleinline
to_json()StochTree::TreeEnsembleinline
TreeEnsemble(int num_trees, int output_dimension=1, bool is_leaf_constant=true, bool is_exponentiated=false)StochTree::TreeEnsembleinline
TreeEnsemble(TreeEnsemble &ensemble)StochTree::TreeEnsembleinline