radial_basis_function.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_ANN_LAYER_RADIAL_BASIS_FUNCTION_HPP
#define MLPACK_METHODS_ANN_LAYER_RADIAL_BASIS_FUNCTION_HPP

#include <mlpack/prereqs.hpp>
#include <mlpack/methods/ann/activation_functions/gaussian_function.hpp>

#include "layer.hpp"

namespace mlpack {
namespace ann  {

template <
    typename MatType = arma::mat,
    typename Activation = GaussianFunction
>
class RBFType : public Layer<MatType>
{
 public:
  RBFType();

  RBFType(const size_t outSize,
          MatType& centres,
          double betas = 0);

  RBFType* Clone() const { return new RBFType(*this); }

  // Virtual destructor.
  virtual ~RBFType() { }

  RBFType(const RBFType& other);
  RBFType(RBFType&& other);
  RBFType& operator=(const RBFType& other);
  RBFType& operator=(RBFType&& other);

  void Forward(const MatType& input, MatType& output);

  void Backward(const MatType& /* input */,
                const MatType& /* gy */,
                MatType& /* g */);

  void ComputeOutputDimensions();

  size_t WeightSize() const { return 0; }

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

 private:
  size_t outSize;

  double betas;

  MatType centres;

  MatType distances;
}; // class RBFType

typedef RBFType<arma::mat> RBF;

} // namespace ann
} // namespace mlpack

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

#endif