MLPACK  1.0.4
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 <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