15 #ifndef __MLPACK_METHODS_DET_DT_UTILS_HPP
16 #define __MLPACK_METHODS_DET_DT_UTILS_HPP
36 const arma::mat& data,
37 const arma::Mat<size_t>& labels,
38 const size_t numClasses,
39 const std::string leafClassMembershipFile =
"");
50 const std::string viFile =
"");
66 const bool useVolumeReg =
false,
67 const size_t maxLeafSize = 10,
68 const size_t minLeafSize = 5,
69 const std::string unprunedTreeOutput =
"");
74 #endif // __MLPACK_METHODS_DET_DT_UTILS_HPP
Linear algebra utility functions, generally performed on matrices or vectors.
void PrintLeafMembership(DTree *dtree, const arma::mat &data, const arma::Mat< size_t > &labels, const size_t numClasses, const std::string leafClassMembershipFile="")
Print the membership of leaves of a density estimation tree given the labels and number of classes...
Include all of the base components required to write MLPACK methods, and the main MLPACK Doxygen docu...
void PrintVariableImportance(const DTree *dtree, const std::string viFile="")
Print the variable importance of each dimension of a density estimation tree.
DTree * Trainer(arma::mat &dataset, const size_t folds, const bool useVolumeReg=false, const size_t maxLeafSize=10, const size_t minLeafSize=5, const std::string unprunedTreeOutput="")
Train the optimal decision tree using cross-validation with the given number of folds.
A density estimation tree is similar to both a decision tree and a space partitioning tree (like a kd...