MLPACK  1.0.8
dt_utils.hpp
Go to the documentation of this file.
1 
23 #ifndef __MLPACK_METHODS_DET_DT_UTILS_HPP
24 #define __MLPACK_METHODS_DET_DT_UTILS_HPP
25 
26 #include <mlpack/core.hpp>
27 #include "dtree.hpp"
28 
29 namespace mlpack {
30 namespace det {
31 
43 void PrintLeafMembership(DTree* dtree,
44  const arma::mat& data,
45  const arma::Mat<size_t>& labels,
46  const size_t numClasses,
47  const std::string leafClassMembershipFile = "");
48 
57 void PrintVariableImportance(const DTree* dtree,
58  const std::string viFile = "");
59 
72 DTree* Trainer(arma::mat& dataset,
73  const size_t folds,
74  const bool useVolumeReg = false,
75  const size_t maxLeafSize = 10,
76  const size_t minLeafSize = 5,
77  const std::string unprunedTreeOutput = "");
78 
79 }; // namespace det
80 }; // namespace mlpack
81 
82 #endif // __MLPACK_METHODS_DET_DT_UTILS_HPP