mlpack
orthogonal_regularizer.hpp
Go to the documentation of this file.
1 
12 #ifndef MLPACK_METHODS_ANN_ORTHOGONAL_REGULARIZER_HPP
13 #define MLPACK_METHODS_ANN_ORTHOGONAL_REGULARIZER_HPP
14 
15 #include <mlpack/prereqs.hpp>
16 
17 namespace mlpack {
18 namespace ann {
19 
38 class OrthogonalRegularizer
39 {
40  public:
46  OrthogonalRegularizer(double factor = 1.0);
47 
55  template<typename MatType>
56  void Evaluate(const MatType& weight, MatType& gradient);
57 
59  template<typename Archive>
60  void serialize(Archive& ar, const uint32_t /* version */);
61 
63  double factor;
64 };
65 
66 } // namespace ann
67 } // namespace mlpack
68 
69 // Include implementation.
70 #include "orthogonal_regularizer_impl.hpp"
71 
72 #endif
mlpack::ann::OrthogonalRegularizer
Implementation of the OrthogonalRegularizer.
Definition: orthogonal_regularizer.hpp:38
mlpack
Linear algebra utility functions, generally performed on matrices or vectors.
Definition: add_to_cli11.hpp:21
mlpack::ann::OrthogonalRegularizer::OrthogonalRegularizer
OrthogonalRegularizer(double factor=1.0)
Create the regularizer object.
prereqs.hpp
The core includes that mlpack expects; standard C++ includes and Armadillo.
mlpack::ann::OrthogonalRegularizer::factor
double factor
The constant for the regularization.
Definition: orthogonal_regularizer.hpp:63
mlpack::ann::OrthogonalRegularizer::Evaluate
void Evaluate(const MatType &weight, MatType &gradient)
Calculate the gradient for regularization.
mlpack::ann::OrthogonalRegularizer::serialize
void serialize(Archive &ar, const uint32_t)
Serialize the regularizer (nothing to do).