StochTree 0.0.1
|
This is the complete list of members for StochTree::TreeEnsemble, including all inherited members.
CloneFromExistingTree(int i, Tree *tree) | StochTree::TreeEnsemble | inline |
from_json(const json &ensemble_json) | StochTree::TreeEnsemble | inline |
GetMaxLeafIndex() | StochTree::TreeEnsemble | inline |
GetTree(int i) | StochTree::TreeEnsemble | inline |
PredictLeafIndices(ForestDataset *dataset) | StochTree::TreeEnsemble | inline |
PredictLeafIndicesInplace(ForestDataset *dataset, std::vector< int32_t > &output, int num_trees, data_size_t n) | StochTree::TreeEnsemble | inline |
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::TreeEnsemble | inline |
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::TreeEnsemble | inline |
PredictLeafIndicesInplace(Eigen::MatrixXd &covariates, std::vector< int32_t > &output, int num_trees, data_size_t n) | StochTree::TreeEnsemble | inline |
ReconstituteFromForest(TreeEnsemble &ensemble) | StochTree::TreeEnsemble | inline |
ResetInitTree(int i) | StochTree::TreeEnsemble | inline |
ResetRoot() | StochTree::TreeEnsemble | inline |
ResetTree(int i) | StochTree::TreeEnsemble | inline |
to_json() | StochTree::TreeEnsemble | inline |
TreeEnsemble(int num_trees, int output_dimension=1, bool is_leaf_constant=true, bool is_exponentiated=false) | StochTree::TreeEnsemble | inline |
TreeEnsemble(TreeEnsemble &ensemble) | StochTree::TreeEnsemble | inline |