13 #ifndef MLPACK_METHODS_SPARSE_CODING_SPARSE_CODING_HPP
14 #define MLPACK_METHODS_SPARSE_CODING_SPARSE_CODING_HPP
25 namespace sparse_coding {
144 template<
typename DictionaryInitializer = DataDependentRandomInitializer>
147 const double lambda1,
148 const double lambda2 = 0,
149 const size_t maxIterations = 0,
150 const double objTolerance = 0.01,
151 const double newtonTolerance = 1e-6,
152 const DictionaryInitializer& initializer =
153 DictionaryInitializer());
172 const double lambda1,
173 const double lambda2 = 0,
174 const size_t maxIterations = 0,
175 const double objTolerance = 0.01,
176 const double newtonTolerance = 1e-6);
181 template<
typename DictionaryInitializer = DataDependentRandomInitializer>
182 void Train(
const arma::mat& data,
183 const DictionaryInitializer& initializer =
184 DictionaryInitializer());
193 void Encode(
const arma::mat& data, arma::mat& codes);
207 const arma::mat& codes,
208 const arma::uvec& adjacencies);
218 double Objective(
const arma::mat& data,
const arma::mat& codes)
const;
226 size_t Atoms()
const {
return atoms; }
256 template<
typename Archive>
257 void Serialize(Archive& ar,
const unsigned int );
264 arma::mat dictionary;
272 size_t maxIterations;
276 double newtonTolerance;
283 #include "sparse_coding_impl.hpp"
double NewtonTolerance() const
Get the tolerance for Newton's method (dictionary optimization step).
double Lambda2() const
Access the L2 regularization term.
void ProjectDictionary()
Project each atom of the dictionary back onto the unit ball, if necessary.
void Encode(const arma::mat &data, arma::mat &codes)
Sparse code each point in the given dataset via LARS, using the current dictionary and store the enco...
The core includes that mlpack expects; standard C++ includes and Armadillo.
const arma::mat & Dictionary() const
Access the dictionary.
double & NewtonTolerance()
Modify the tolerance for Newton's method (dictionary optimization step).
double & ObjTolerance()
Modify the objective tolerance.
double & Lambda1()
Modify the L1 regularization term.
size_t & Atoms()
Modify the number of atoms.
double Objective(const arma::mat &data, const arma::mat &codes) const
Compute the objective function.
An implementation of Sparse Coding with Dictionary Learning that achieves sparsity via an l1-norm reg...
size_t & MaxIterations()
Modify the maximum number of iterations.
size_t Atoms() const
Access the number of atoms.
SparseCoding(const arma::mat &data, const size_t atoms, const double lambda1, const double lambda2=0, const size_t maxIterations=0, const double objTolerance=0.01, const double newtonTolerance=1e-6, const DictionaryInitializer &initializer=DictionaryInitializer())
Set the parameters to SparseCoding.
size_t MaxIterations() const
Get the maximum number of iterations.
void Train(const arma::mat &data, const DictionaryInitializer &initializer=DictionaryInitializer())
Train the sparse coding model on the given dataset.
arma::mat & Dictionary()
Modify the dictionary.
double Lambda1() const
Access the L1 regularization term.
double & Lambda2()
Modify the L2 regularization term.
void Serialize(Archive &ar, const unsigned int)
Serialize the sparse coding model.
double ObjTolerance() const
Get the objective tolerance.
double OptimizeDictionary(const arma::mat &data, const arma::mat &codes, const arma::uvec &adjacencies)
Learn dictionary via Newton method based on Lagrange dual.