MLPACK
1.0.4
|
00001 00023 #ifndef __MLPACK_METHODS_DET_DT_UTILS_HPP 00024 #define __MLPACK_METHODS_DET_DT_UTILS_HPP 00025 00026 #include <string> 00027 00028 #include <mlpack/core.hpp> 00029 #include "dtree.hpp" 00030 00031 namespace mlpack { 00032 namespace det { 00033 00045 void PrintLeafMembership(DTree* dtree, 00046 const arma::mat& data, 00047 const arma::Mat<size_t>& labels, 00048 const size_t numClasses, 00049 const std::string leafClassMembershipFile = ""); 00050 00059 void PrintVariableImportance(const DTree* dtree, 00060 const std::string viFile = ""); 00061 00073 DTree* Trainer(arma::mat& dataset, 00074 const size_t folds, 00075 const bool useVolumeReg = false, 00076 const size_t maxLeafSize = 10, 00077 const size_t minLeafSize = 5, 00078 const std::string unprunedTreeOutput = ""); 00079 00080 }; // namespace det 00081 }; // namespace mlpack 00082 00083 #endif // __MLPACK_METHODS_DET_DT_UTILS_HPP