blob: 83e63121dc1a406ab40a8b28174c23260f9cd73d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
#ifndef LIN1D_ESTIMATER
#define LIN1D_ESTIMATER
#include <core/pre_estimater.hpp>
#include <misc/optvec.hpp>
#include <models/lin1d.hpp>
#include <vector>
namespace opt_utilities
{
template <typename T>
class lin1d_estimater
:public pre_estimater<T,T,std::vector<T>,std::string>
{
public:
lin1d_estimater()
{
this->set_model_id("1d linear model");
}
lin1d_estimater<T>* do_clone()const
{
return new lin1d_estimater<T>(*this);
}
void do_estimate(const data_set<T,T>& d,model<T,T,std::vector<T>,std::string>& m)const
{
T n=d.size();
T sy=0;
T sxx=0;
T sx=0;
T sxy=0;
for(int i=0;i<d.size();++i)
{
sy+=d.get_data(i).get_y();
sxx+=d.get_data(i).get_x()*d.get_data(i).get_x();
sx+=d.get_data(i).get_x();
sxy+=d.get_data(i).get_x()*d.get_data(i).get_y();
}
T b=(sy*sxx-sx*sxy)/(n*sxx-sx*sx);
T k=(n*sxy-sx*sy)/(n*sxx-sx*sx);
m.set_param_value("k",k);
m.set_param_value("b",b);
}
};
}
#endif
|
