dt_utils.hpp
Go to the documentation of this file.00001
00023 #ifndef __MLPACK_METHODS_DET_DT_UTILS_HPP
00024 #define __MLPACK_METHODS_DET_DT_UTILS_HPP
00025
00026 #include <mlpack/core.hpp>
00027 #include "dtree.hpp"
00028
00029 namespace mlpack {
00030 namespace det {
00031
00043 void PrintLeafMembership(DTree* dtree,
00044 const arma::mat& data,
00045 const arma::Mat<size_t>& labels,
00046 const size_t numClasses,
00047 const std::string leafClassMembershipFile = "");
00048
00057 void PrintVariableImportance(const DTree* dtree,
00058 const std::string viFile = "");
00059
00072 DTree* Trainer(arma::mat& dataset,
00073 const size_t folds,
00074 const bool useVolumeReg = false,
00075 const size_t maxLeafSize = 10,
00076 const size_t minLeafSize = 5,
00077 const std::string unprunedTreeOutput = "");
00078
00079 };
00080 };
00081
00082 #endif // __MLPACK_METHODS_DET_DT_UTILS_HPP