PCA.h

Go to the documentation of this file.

#ifndef OPENANN_PCA_H_
#define OPENANN_PCA_H_

#include <OpenANN/Transformer.h>
#include <Eigen/Core>

namespace OpenANN
{

class PCA : public Transformer
{
  int components;
  bool whiten;
  Eigen::VectorXd mean;
  Eigen::MatrixXd W;
  Eigen::VectorXd evr;
public:
  PCA(int components, bool whiten = true);

  virtual Transformer& fit(const Eigen::MatrixXd& X);
  virtual Eigen::MatrixXd transform(const Eigen::MatrixXd& X);

  Eigen::VectorXd explainedVarianceRatio();
};

} // namespace OpenANN

#endif // OPENANN_PCA_H_