pca.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_PCA_PCA_HPP
#define MLPACK_METHODS_PCA_PCA_HPP

#include <mlpack/prereqs.hpp>
#include <mlpack/methods/pca/decomposition_policies/exact_svd_method.hpp>

namespace mlpack {
namespace pca {

template<typename DecompositionPolicy = ExactSVDPolicy>
class PCA
{
 public:
  PCA(const bool scaleData = false,
      const DecompositionPolicy& decomposition = DecompositionPolicy());

  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(const arma::mat& data,
             arma::mat& transformedData);

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

  inline double Apply(arma::mat& data, const int newDimension)
  {
    return Apply(data, size_t(newDimension));
  }

  double Apply(arma::mat& data, const double varRetained);

  bool ScaleData() const { return scaleData; }
  bool& ScaleData() { return scaleData; }

 private:
  void ScaleData(arma::mat& centeredData)
  {
    if (scaleData)
    {
      // Scaling the data is when we reduce the variance of each dimension
      // to 1. We do this by dividing each dimension by its standard
      // deviation.
      arma::vec stdDev = arma::stddev(
          centeredData, 0, 1 /* for each dimension */);

      // If there are any zeroes, make them very small.
      for (size_t i = 0; i < stdDev.n_elem; ++i)
        if (stdDev[i] == 0)
          stdDev[i] = 1e-50;

      centeredData /= arma::repmat(stdDev, 1, centeredData.n_cols);
    }
  }

  bool scaleData;

  DecompositionPolicy decomposition;
}; // class PCA

} // namespace pca
} // namespace mlpack

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

#endif