mirror of
https://github.com/aicodix/dsp.git
synced 2026-04-27 14:30:36 +00:00
71 lines
1.3 KiB
C++
71 lines
1.3 KiB
C++
/*
|
|
Regression analysis
|
|
|
|
Copyright 2018 Ahmet Inan <inan@aicodix.de>
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
namespace DSP {
|
|
|
|
template <typename TYPE>
|
|
class SimpleLinearRegression
|
|
{
|
|
TYPE xint_, yint_, slope_;
|
|
public:
|
|
SimpleLinearRegression() : xint_(0), yint_(0), slope_(0) {}
|
|
SimpleLinearRegression(const TYPE *x, const TYPE *y, int LEN)
|
|
{
|
|
TYPE avgX(0), avgY(0);
|
|
for (int i = 0; i < LEN; ++i) {
|
|
avgX += x[i];
|
|
avgY += y[i];
|
|
}
|
|
avgX /= LEN;
|
|
avgY /= LEN;
|
|
TYPE varX(0), covXY(0);
|
|
for (int i = 0; i < LEN; ++i) {
|
|
varX += (x[i] - avgX) * (x[i] - avgX);
|
|
covXY += (x[i] - avgX) * (y[i] - avgY);
|
|
}
|
|
xint_ = avgX - avgY * varX / covXY;
|
|
slope_ = covXY / varX;
|
|
yint_ = avgY - slope_ * avgX;
|
|
}
|
|
SimpleLinearRegression(const TYPE *y, int l, TYPE x0 = 0, TYPE dx = 1)
|
|
{
|
|
TYPE midL((l-1) / TYPE(2));
|
|
TYPE avgX(x0 + dx*midL);
|
|
TYPE avgY(0);
|
|
for (int i = 0; i < l; ++i)
|
|
avgY += y[i];
|
|
avgY /= l;
|
|
TYPE varX(0), covXY(0);
|
|
for (int i = 0; i < l; ++i) {
|
|
varX += (dx * (i - midL)) * (dx * (i - midL));
|
|
covXY += (dx * (i - midL)) * (y[i] - avgY);
|
|
}
|
|
xint_ = avgX - avgY * varX / covXY;
|
|
slope_ = covXY / varX;
|
|
yint_ = avgY - slope_ * avgX;
|
|
}
|
|
TYPE xint()
|
|
{
|
|
return xint_;
|
|
}
|
|
TYPE slope()
|
|
{
|
|
return slope_;
|
|
}
|
|
TYPE yint()
|
|
{
|
|
return yint_;
|
|
}
|
|
TYPE operator () (TYPE x)
|
|
{
|
|
return yint_ + slope_ * x;
|
|
}
|
|
};
|
|
|
|
}
|
|
|