mlpack
|
The model to save to disk. More...
#include <adaboost_model.hpp>
Public Types | |
enum | WeakLearnerTypes { DECISION_STUMP, PERCEPTRON } |
Public Member Functions | |
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 & | operator= (const AdaBoostModel &other) |
Copy assignment operator. | |
AdaBoostModel & | operator= (AdaBoostModel &&other) |
Move assignment operator. | |
~AdaBoostModel () | |
Clean up memory. | |
const arma::Col< size_t > & | Mappings () const |
Get the mappings. | |
arma::Col< size_t > & | Mappings () |
Modify the mappings. | |
size_t | WeakLearnerType () const |
Get the weak learner type. | |
size_t & | WeakLearnerType () |
Modify the weak learner type. | |
size_t | Dimensionality () const |
Get the dimensionality of the model. | |
size_t & | Dimensionality () |
Modify the dimensionality of the model. | |
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... | |
void | Classify (const arma::mat &testData, arma::Row< size_t > &predictions) |
Classify test points. | |
void | Classify (const arma::mat &testData, arma::Row< size_t > &predictions, arma::mat &probabilities) |
Classify test points. | |
template<typename Archive > | |
void | serialize (Archive &ar, const uint32_t) |
Serialize the model. | |
The model to save to disk.
void AdaBoostModel::Train | ( | const arma::mat & | data, |
const arma::Row< size_t > & | labels, | ||
const size_t | numClasses, | ||
const size_t | iterations, | ||
const double | tolerance | ||
) |