Normalization.h

Go to the documentation of this file.

#ifndef OPENANN_NORMALIZATION_H_
#define OPENANN_NORMALIZATION_H_

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

namespace OpenANN
{

class Normalization : public Transformer
{
  Eigen::MatrixXd mean;
  Eigen::MatrixXd std;
public:
  Normalization();

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

  Eigen::VectorXd getMean();
  Eigen::VectorXd getStd();
};

} // namespace OpenANN

#endif // OPENANN_NORMALIZATION_H_