14 #ifndef __MLPACK_METHODS_PERCEPTRON_PERCEPTRON_HPP 15 #define __MLPACK_METHODS_PERCEPTRON_PERCEPTRON_HPP 24 namespace perceptron {
35 template<
typename LearnPolicy = SimpleWeightUpdate,
36 typename WeightInitializationPolicy = ZeroInitialization,
37 typename MatType = arma::mat>
52 const size_t dimensionality = 0,
69 const arma::Row<size_t>& labels,
70 const size_t numClasses,
85 const arma::Row<size_t>& labels,
86 const arma::rowvec& instanceWeights);
103 void Train(
const MatType& data,
104 const arma::Row<size_t>& labels,
105 const arma::rowvec& instanceWeights = arma::rowvec());
115 void Classify(
const MatType& test, arma::Row<size_t>& predictedLabels);
120 template<
typename Archive>
121 void Serialize(Archive& ar,
const unsigned int );
160 #include "perceptron_impl.hpp" arma::mat weights
Stores the weights for each of the input class labels.
arma::vec biases
The biases for each class.
Linear algebra utility functions, generally performed on matrices or vectors.
const arma::vec & Biases() const
Get the biases.
size_t NumClasses() const
Get the number of classes this perceptron has been trained for.
Perceptron(const size_t numClasses=0, const size_t dimensionality=0, const size_t maxIterations=1000)
Constructor: create the perceptron with the given number of classes and initialize the weight matrix...
arma::vec & Biases()
Modify the biases. You had better know what you are doing!
const arma::mat & Weights() const
Get the weight matrix.
size_t maxIterations
The maximum number of iterations during training.
size_t & MaxIterations()
Modify the maximum number of iterations.
size_t MaxIterations() const
Get the maximum number of iterations.
Include all of the base components required to write MLPACK methods, and the main MLPACK Doxygen docu...
void Serialize(Archive &ar, const unsigned int)
Serialize the perceptron.
void Train(const MatType &data, const arma::Row< size_t > &labels, const arma::rowvec &instanceWeights=arma::rowvec())
Train the perceptron on the given data for up to the maximum number of iterations (specified in the c...
void Classify(const MatType &test, arma::Row< size_t > &predictedLabels)
Classification function.
arma::mat & Weights()
Modify the weight matrix. You had better know what you are doing!
This class implements a simple perceptron (i.e., a single layer neural network).