22 #ifndef __MLPACK_METHODS_DECISION_STUMP_DECISION_STUMP_HPP
23 #define __MLPACK_METHODS_DECISION_STUMP_DECISION_STUMP_HPP
28 namespace decision_stump {
43 template <
typename MatType = arma::mat>
57 const arma::Row<size_t>& labels,
59 size_t inpBucketSize);
69 void Classify(
const MatType& test, arma::Row<size_t>& predictedLabels);
128 const arma::Row<size_t>& labels);
137 template <
typename rType>
void TrainOnAtt(
const arma::rowvec& attribute,
138 const arma::Row<size_t>& labels);
152 template <
typename rType> rType
CountMostFreq(
const arma::Row<rType>& subCols);
159 template <
typename rType>
int IsDistinct(
const arma::Row<rType>& featureRow);
167 template <
typename AttType,
typename LabelType>
174 #include "decision_stump_impl.hpp"