ZCAWhitening.h

Go to the documentation of this file.

#ifndef OPENANN_ZCA_WHITENING_H_
#define OPENANN_ZCA_WHITENING_H_

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

namespace OpenANN
{

class ZCAWhitening : public Transformer
{
  Eigen::VectorXd mean;
  Eigen::MatrixXd W;
public:
  virtual Transformer& fit(const Eigen::MatrixXd& X);
  virtual Eigen::MatrixXd transform(const Eigen::MatrixXd& X);
};

} // namespace OpenANN

#endif // OPENANN_ZCA_WHITENING_H_