radial_basis_function_impl.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_ANN_LAYER_RBF_IMPL_HPP
#define MLPACK_METHODS_ANN_LAYER_RBF_IMPL_HPP

// In case it hasn't yet been included.
#include "radial_basis_function.hpp"

namespace mlpack {
namespace ann  {

template<typename InputDataType, typename OutputDataType,
         typename Activation>
RBF<InputDataType, OutputDataType, Activation>::RBF() :
    inSize(0),
    outSize(0),
    sigmas(0),
    betas(0)
{
  // Nothing to do here.
}

template<typename InputDataType, typename OutputDataType,
         typename Activation>
RBF<InputDataType, OutputDataType, Activation>::RBF(
    const size_t inSize,
    const size_t outSize,
    arma::mat& centres,
    double betas) :
    inSize(inSize),
    outSize(outSize),
    betas(betas),
    centres(centres)
{
  sigmas = 0;
  if (betas == 0)
  {
    for (size_t i = 0; i < centres.n_cols; i++)
    {
      double max_dis = 0;
      arma::mat temp = centres.each_col() - centres.col(i);
      max_dis = arma::accu(arma::max(arma::pow(arma::sum(
          arma::pow((temp), 2), 0), 0.5).t()));
      if (max_dis > sigmas)
        sigmas = max_dis;
    }
    this->betas = std::pow(2 * outSize, 0.5) / sigmas;
  }
}

template<typename InputDataType, typename OutputDataType,
         typename Activation>
template<typename eT>
void RBF<InputDataType, OutputDataType, Activation>::Forward(
    const arma::Mat<eT>& input,
    arma::Mat<eT>& output)
{
  distances = arma::mat(outSize, input.n_cols);

  for (size_t i = 0; i < input.n_cols; i++)
  {
    arma::mat temp = centres.each_col() - input.col(i);
    distances.col(i) = arma::pow(arma::sum(
      arma::pow((temp), 2), 0), 0.5).t();
  }
  Activation::Fn(distances * std::pow(betas, 0.5),
      output);
}


template<typename InputDataType, typename OutputDataType,
         typename Activation>
template<typename eT>
void RBF<InputDataType, OutputDataType, Activation>::Backward(
    const arma::Mat<eT>& /* input */,
    const arma::Mat<eT>& /* gy */,
    arma::Mat<eT>& /* g */)
{
  // Nothing to do here.
}

template<typename InputDataType, typename OutputDataType,
         typename Activation>
template<typename Archive>
void RBF<InputDataType, OutputDataType, Activation>::serialize(
    Archive& ar,
    const uint32_t /* version */)
{
  ar(CEREAL_NVP(distances));
  ar(CEREAL_NVP(centres));
}

} // namespace ann
} // namespace mlpack

#endif