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.