aboutsummaryrefslogtreecommitdiffstats
path: root/pre_estimaters/vdualgauss1d_estimater.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'pre_estimaters/vdualgauss1d_estimater.hpp')
-rw-r--r--pre_estimaters/vdualgauss1d_estimater.hpp73
1 files changed, 73 insertions, 0 deletions
diff --git a/pre_estimaters/vdualgauss1d_estimater.hpp b/pre_estimaters/vdualgauss1d_estimater.hpp
new file mode 100644
index 0000000..2be4cfb
--- /dev/null
+++ b/pre_estimaters/vdualgauss1d_estimater.hpp
@@ -0,0 +1,73 @@
+#ifndef VDUALGAUSS1D_ESTIMATER
+#define VDUALGAUSS1D_ESTIMATER
+#include <core/pre_estimater.hpp>
+#include <misc/optvec.hpp>
+#include <vmodels/dualgauss1d.hpp>
+#include <vector>
+
+namespace opt_utilities
+{
+ template <typename T>
+ class dualgauss1d_estimater
+ :public pre_estimater<optvec<T>,optvec<T>,optvec<T>,std::string>
+ {
+ public:
+ dualgauss1d_estimater()
+ {
+ this->set_model_id("1d dualgaussian");
+ }
+
+ dualgauss1d_estimater<T>* do_clone()const
+ {
+ return new dualgauss1d_estimater<T>(*this);
+ }
+
+ void do_estimate(const data_set<optvec<T>,optvec<T> >& d,model<optvec<T>,optvec<T>,optvec<T>,std::string>& m)const
+ {
+ int n=d.size();
+
+ T xp1st=0;
+ T xp2nd=0;
+ T yp1st=d.get_data(0).get_y()[0];
+ T yp2nd=d.get_data(0).get_y()[0];
+
+ for(int i=0;i<n;++i)
+ {
+ if(d.get_data(i).get_y()[0]>yp1st)
+ {
+ yp1st=d.get_data(i).get_y()[0];
+ xp1st=d.get_data(i).get_x()[0];
+ }
+ }
+
+ for(int i=0;i<n;++i)
+ {
+
+ }
+
+
+ T xmean=0;
+ T x2mean=0;
+ T wgt=0;
+ T wgt2=0;
+ for(int i=0;i<n;++i)
+ {
+ T x=d.get_data(i).get_x()[0];
+ T y=d.get_data(i).get_y()[0];
+ xmean+=x*y;
+ x2mean+=x*x*y;
+ wgt+=y;
+ }
+ xmean/=wgt;
+ x2mean/=wgt;
+ T sigma=std::sqrt(x2mean-xmean*xmean);
+ m.set_param_value("x0",xmean);
+ m.set_param_value("sigma",sigma);
+
+ }
+ };
+}
+
+
+
+#endif