lregularizer.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_ANN_LREGULARIZER_HPP
#define MLPACK_METHODS_ANN_LREGULARIZER_HPP

#include <mlpack/prereqs.hpp>

namespace mlpack {
namespace ann {

template<int TPower>
class LRegularizer
{
 public:
  LRegularizer(double factor = 1.0);

  template<typename MatType>
  void Evaluate(const MatType& weight, MatType& gradient);

  template<typename Archive>
  void serialize(Archive& ar, const unsigned int /* version */);

  static const int Power = TPower;

  double factor;
};

// Convenience typedefs.
typedef LRegularizer<1> L1Regularizer;

typedef LRegularizer<2> L2Regularizer;

} // namespace ann
} // namespace mlpack

// Include implementation.
#include "lregularizer_impl.hpp"

#endif