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);
85 const arma::rowvec& weights,
86 const arma::Row<size_t>& labels);
127 template <
bool isWeight>
129 const arma::Row<size_t>& labels,
130 const arma::rowvec& weightD);
139 template <
typename rType>
void TrainOnAtt(
const arma::rowvec& attribute,
140 const arma::Row<size_t>& labels);
154 template <
typename rType> rType
CountMostFreq(
const arma::Row<rType>&
162 template <
typename rType>
int IsDistinct(
const arma::Row<rType>& featureRow);
171 template <
typename LabelType,
bool isWeight>
173 const arma::rowvec& tempD);
182 template <
bool isWeight>
183 void Train(
const MatType& data,
const arma::Row<size_t>& labels,
184 const arma::rowvec& weightD);
191 #include "decision_stump_impl.hpp"