CscSupervisedLearningRegression.h

Go to the documentation of this file.

/*
 * CscSupervisedLearningRegression.h
 *
 *  Created on: 28 Juin 2023
 *      Author: Max Lhermitte
 */
 
#ifndef CscSupervisedLearningRegression_h
#define CscSupervisedLearningRegression_h
 
#include "CscCommon.h"
#include <Eigen/Dense>
#include <fstream>
#include <iostream>
using std::vector, std::string;
 
class CSC_DLL_IMPORTEXPORT CscSupervisedLearningRegression {
public:
    struct CSC_DLL_IMPORTEXPORT RegressionInputsData {
    public:
        vector<vector<float>> data;
        const vector<string> namesVar;
 
        RegressionInputsData(vector<vector<float>> data, vector<string> namesVar) : data(data),
                                                                                    namesVar(namesVar) {
        }
        vector<vector<float>> getInputsData() const {
            return data;
        }
        vector<string> getNamesVar() const {
            return namesVar;
        }
        void addInputsData(vector<float> dataToAdd) {
            data.push_back(dataToAdd);
        }
    };
 
    struct CSC_DLL_IMPORTEXPORT RegressionOutputsData {
    public:
        double r2; // coefficient of dertermination
        vector<float> polynomialCoefficients;
 
        RegressionOutputsData(double r2, vector<float> polynomialCoefficients) : r2(r2),
                                                                                 polynomialCoefficients(polynomialCoefficients) {
        }
    };
 
    CscSupervisedLearningRegression(RegressionInputsData regressionInputsData);
 
    RegressionOutputsData polynomialRegression(const int degree, bool dataToCsv = false);
 
    RegressionOutputsData bestDegreePolynomialRegression(bool saveToCsv = false);
 
    static float findMaxLocal(const vector<float> &coeffPolynomial, float minBounding, float maxBounding, float precision);
 
private:
    RegressionInputsData regressionInputsData;
 
    static Eigen::MatrixXf polyTransform(const Eigen::MatrixXf &data, int degree);
 
    static double calculCoefOfDetermination(const Eigen::MatrixXf &inputs, const Eigen::MatrixXf &response);
 
    void saveToCsv(const string &filename);
 
    unique_ptr<CscLogger> logger = CscLogger::getForCategory("CscSupervisedLearningRegression");
};
 
#endif