MLPACK
1.0.8
Main Page
Related Pages
Namespaces
Classes
Files
File List
File Members
src
mlpack
methods
det
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
Generated by
1.8.3.1