mirror of
https://github.com/Relintai/pmlpp.git
synced 2024-11-13 13:57:19 +01:00
35 lines
774 B
C++
35 lines
774 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<real_t> x, std::vector<real_t> y) :
|
|
inputSet(x), outputSet(y) {
|
|
MLPPStat estimator;
|
|
b1 = estimator.b1Estimation(inputSet, outputSet);
|
|
b0 = estimator.b0Estimation(inputSet, outputSet);
|
|
}
|
|
|
|
std::vector<real_t> MLPPUniLinReg::modelSetTest(std::vector<real_t> x) {
|
|
MLPPLinAlg alg;
|
|
return alg.scalarAdd(b0, alg.scalarMultiply(b1, x));
|
|
}
|
|
|
|
real_t MLPPUniLinReg::modelTest(real_t input) {
|
|
return b0 + b1 * input;
|
|
}
|