orthogonal_regularizer.hpp

Go to the documentation of this file.

 
#ifndef MLPACK_METHODS_ANN_ORTHOGONAL_REGULARIZER_HPP
#define MLPACK_METHODS_ANN_ORTHOGONAL_REGULARIZER_HPP
 
#include <mlpack/prereqs.hpp>
 
namespace mlpack {
namespace ann {
 
class OrthogonalRegularizer
{
 public:
  OrthogonalRegularizer(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 */);
 
  double factor;
};
 
} // namespace ann
} // namespace mlpack
 
// Include implementation.
#include "orthogonal_regularizer_impl.hpp"
 
#endif