The model to save to disk. More...
Public Types | |
enum | WeakLearnerTypes { DECISION_STUMP , PERCEPTRON } |
Public Member Functions | |
AdaBoostModel () | |
Create an empty AdaBoost model. More... | |
AdaBoostModel (const arma::Col< size_t > &mappings, const size_t weakLearnerType) | |
Create the AdaBoost model with the given mappings and type. More... | |
AdaBoostModel (const AdaBoostModel &other) | |
Copy constructor. More... | |
AdaBoostModel (AdaBoostModel &&other) | |
Move constructor. More... | |
~AdaBoostModel () | |
Clean up memory. More... | |
void | Classify (const arma::mat &testData, arma::Row< size_t > &predictions) |
Classify test points. More... | |
void | Classify (const arma::mat &testData, arma::Row< size_t > &predictions, arma::mat &probabilities) |
Classify test points. More... | |
size_t | Dimensionality () const |
Get the dimensionality of the model. More... | |
size_t & | Dimensionality () |
Modify the dimensionality of the model. More... | |
const arma::Col< size_t > & | Mappings () const |
Get the mappings. More... | |
arma::Col< size_t > & | Mappings () |
Modify the mappings. More... | |
AdaBoostModel & | operator= (const AdaBoostModel &other) |
Copy assignment operator. More... | |
AdaBoostModel & | operator= (AdaBoostModel &&other) |
Move assignment operator. More... | |
template < typename Archive > | |
void | serialize (Archive &ar, const uint32_t) |
Serialize the model. More... | |
void | Train (const arma::mat &data, const arma::Row< size_t > &labels, const size_t numClasses, const size_t iterations, const double tolerance) |
Train the model, treat the data is all of the numeric type. More... | |
size_t | WeakLearnerType () const |
Get the weak learner type. More... | |
size_t & | WeakLearnerType () |
Modify the weak learner type. More... | |
The model to save to disk.
Definition at line 26 of file adaboost_model.hpp.
enum WeakLearnerTypes |
Enumerator | |
---|---|
DECISION_STUMP | |
PERCEPTRON |
Definition at line 29 of file adaboost_model.hpp.
AdaBoostModel | ( | ) |
Create an empty AdaBoost model.
AdaBoostModel | ( | const arma::Col< size_t > & | mappings, |
const size_t | weakLearnerType | ||
) |
Create the AdaBoost model with the given mappings and type.
AdaBoostModel | ( | const AdaBoostModel & | other | ) |
Copy constructor.
AdaBoostModel | ( | AdaBoostModel && | other | ) |
Move constructor.
~AdaBoostModel | ( | ) |
Clean up memory.
void Classify | ( | const arma::mat & | testData, |
arma::Row< size_t > & | predictions | ||
) |
Classify test points.
Referenced by AdaBoostModel::Dimensionality().
void Classify | ( | const arma::mat & | testData, |
arma::Row< size_t > & | predictions, | ||
arma::mat & | probabilities | ||
) |
Classify test points.
|
inline |
Get the dimensionality of the model.
Definition at line 81 of file adaboost_model.hpp.
|
inline |
Modify the dimensionality of the model.
Definition at line 83 of file adaboost_model.hpp.
References AdaBoostModel::Classify(), and AdaBoostModel::Train().
|
inline |
Get the mappings.
Definition at line 71 of file adaboost_model.hpp.
|
inline |
Modify the mappings.
Definition at line 73 of file adaboost_model.hpp.
AdaBoostModel& operator= | ( | const AdaBoostModel & | other | ) |
Copy assignment operator.
AdaBoostModel& operator= | ( | AdaBoostModel && | other | ) |
Move assignment operator.
|
inline |
void Train | ( | const arma::mat & | data, |
const arma::Row< size_t > & | labels, | ||
const size_t | numClasses, | ||
const size_t | iterations, | ||
const double | tolerance | ||
) |
Train the model, treat the data is all of the numeric type.
Referenced by AdaBoostModel::Dimensionality().
|
inline |
Get the weak learner type.
Definition at line 76 of file adaboost_model.hpp.
|
inline |
Modify the weak learner type.
Definition at line 78 of file adaboost_model.hpp.