kernel_pca.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_KERNEL_PCA_KERNEL_PCA_HPP
#define MLPACK_METHODS_KERNEL_PCA_KERNEL_PCA_HPP

#include <mlpack/prereqs.hpp>
#include <mlpack/methods/kernel_pca/kernel_rules/naive_method.hpp>

namespace mlpack {
namespace kpca {

template <
  typename KernelType,
  typename KernelRule = NaiveKernelRule<KernelType>
>
class KernelPCA
{
 public:
  KernelPCA(const KernelType kernel = KernelType(),
            const bool centerTransformedData = false);

  void Apply(const arma::mat& data,
             arma::mat& transformedData,
             arma::vec& eigval,
             arma::mat& eigvec,
             const size_t newDimension);

  void Apply(const arma::mat& data,
             arma::mat& transformedData,
             arma::vec& eigval,
             arma::mat& eigvec);

  void Apply(const arma::mat& data,
             arma::mat& transformedData,
             arma::vec& eigval);

  void Apply(arma::mat& data, const size_t newDimension);

  const KernelType& Kernel() const { return kernel; }
  KernelType& Kernel() { return kernel; }

  bool CenterTransformedData() const { return centerTransformedData; }
  bool& CenterTransformedData() { return centerTransformedData; }

 private:
  KernelType kernel;
  bool centerTransformedData;
}; // class KernelPCA

} // namespace kpca
} // namespace mlpack

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

#endif // MLPACK_METHODS_KERNEL_PCA_KERNEL_PCA_HPP