hyperbolic_tangent_kernel.hpp

Go to the documentation of this file.

 
#ifndef MLPACK_CORE_KERNELS_HYPERBOLIC_TANGENT_KERNEL_HPP
#define MLPACK_CORE_KERNELS_HYPERBOLIC_TANGENT_KERNEL_HPP
 
#include <mlpack/prereqs.hpp>
 
namespace mlpack {
namespace kernel {
 
class HyperbolicTangentKernel
{
 public:
  HyperbolicTangentKernel() : scale(1.0), offset(0.0)
  { }
 
  HyperbolicTangentKernel(double scale, double offset) :
      scale(scale), offset(offset)
  { }
 
  template<typename VecTypeA, typename VecTypeB>
  double Evaluate(const VecTypeA& a, const VecTypeB& b)
  {
    return tanh(scale * arma::dot(a, b) + offset);
  }
 
  double Scale() const { return scale; }
  double& Scale() { return scale; }
 
  double Offset() const { return offset; }
  double& Offset() { return offset; }
 
  template<typename Archive>
  void serialize(Archive& ar, const unsigned int /* version */)
  {
    ar & BOOST_SERIALIZATION_NVP(scale);
    ar & BOOST_SERIALIZATION_NVP(offset);
  }
 
 private:
  double scale;
  double offset;
};
 
} // namespace kernel
} // namespace mlpack
 
#endif