pcatransformer.h

/*
 * This file is part of Copiste.
 *
 * Copiste is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 *  Copiste is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with Copiste.  If not, see <http://www.gnu.org/licenses/>.
 */


#ifndef INCLUDED_PCATRANSFORMERH
#define INCLUDED_PCATRANSFORMERH

class PCATransformer;

#include <Eigen/Dense>
#include <sstream>

#include "algo/corpus.h"

using namespace Eigen;

class PCATransformer
{
    public:
        PCATransformer(bool debug = false);

        void computePCA(Corpus* corpus);

        bool load(string filename);

        bool save(string filename);

        vector<float> transform(vector<float> input);

        Corpus* transform(const Corpus &corpus, int coord = 2);

        static string serializeMatrix(MatrixXf m);

        static MatrixXf readMatrix(string m, int nbRows = 1, int nbCols = 1);


    private:
        VectorXf mEigenValues;
        MatrixXf mEigenVectors;
        MatrixXf mTransform;
        MatrixXf s;

        VectorXf mMean;

        bool mDebug;
};

#endif