PCA.h

Go to the documentation of this file.

#pragma once
 
#include <cassert>
#include <cmath>
#include <iostream>
#include <vector>
#include "../krotos_eigen_updated/Eigen/Dense"
 
#ifndef JUCE_WINDOWS
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Weverything"
#else
__pragma(warning(push)) __pragma(warning(disable : 4127 4267))
#endif
 
#ifdef JUCE_MAC
#pragma GCC diagnostic pop
#else
        __pragma(warning(pop))
#endif
 
namespace krotos
{
 
class PCA
{
 
  public:
    PCA();
    ~PCA();
 
    void reserve(const std::size_t& n);
    void insert(const std::vector<float>& x);
    void fit(const std::size_t& n_components);
    std::vector<float> getPrincipalComponents(const std::size_t& i);
    void clear();
    void toCSV(const std::string& filename);
 
  private:
    std::size_t rows = 0;
    std::vector<float> data;
    std::vector<float> principal_components;
    bool is_fitted = false;
};
 
} // namespace krotos