pmlpp/mlpp/uni_lin_reg/uni_lin_reg.cpp

35 lines
775 B
C++

//
// UniLinReg.cpp
//
// Created by Marc Melikyan on 9/29/20.
//
#include "uni_lin_reg.h"
#include "../lin_alg/lin_alg.h"
#include "../stat/stat.h"
#include <iostream>
// General Multivariate Linear Regression Model
// ŷ = b0 + b1x1 + b2x2 + ... + bkxk
// Univariate Linear Regression Model
// ŷ = b0 + b1x1
MLPPUniLinReg::MLPPUniLinReg(std::vector<double> x, std::vector<double> y) :
inputSet(x), outputSet(y) {
MLPPStat estimator;
b1 = estimator.b1Estimation(inputSet, outputSet);
b0 = estimator.b0Estimation(inputSet, outputSet);
}
std::vector<double> MLPPUniLinReg::modelSetTest(std::vector<double> x) {
MLPPLinAlg alg;
return alg.scalarAdd(b0, alg.scalarMultiply(b1, x));
}
double MLPPUniLinReg::modelTest(double input) {
return b0 + b1 * input;
}