Compressor.h

Go to the documentation of this file.

#ifndef OPENANN_COMPRESSOR_H_
#define OPENANN_COMPRESSOR_H_

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

namespace OpenANN
{

class Compressor : public Transformer
{
  Eigen::MatrixXd cm;
public:
  Compressor(int inputDim, int outputDim,
             CompressionMatrixFactory::Transformation transformation);
  virtual Transformer& fit(const Eigen::MatrixXd& X);
  virtual Transformer& fitPartial(const Eigen::MatrixXd& X);
  virtual Eigen::MatrixXd transform(const Eigen::MatrixXd& X);
  int getOutputs();
};

} // namespace OpenANN

#endif // OPENANN_COMPRESSOR_H_