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 }; // namespace det
00080 }; // namespace mlpack
00081 
00082 #endif // __MLPACK_METHODS_DET_DT_UTILS_HPP

Generated on 13 Aug 2014 for MLPACK by  doxygen 1.6.1