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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
#ifndef CONSTRAINED_DBETA
#define CONSTRAINED_DBETA
#include "projector.hpp"
#include <iostream>
namespace opt_utilities
{
template <typename T>
class constrained_dbeta
:public model<std::vector<T>,std::vector<T>,std::vector<T> >
{
public:
constrained_dbeta()
{
this->push_param_info(param_info<std::vector<T>,std::string>("n01",1));
this->push_param_info(param_info<std::vector<T>,std::string>("beta1",.66));
this->push_param_info(param_info<std::vector<T>,std::string>("rc1",100));
this->push_param_info(param_info<std::vector<T>,std::string>("n02",1));
this->push_param_info(param_info<std::vector<T>,std::string>("beta2",.67));
this->push_param_info(param_info<std::vector<T>,std::string>("rc2",110));
}
public:
constrained_dbeta<T>* do_clone()const
{
return new constrained_dbeta<T>(*this);
}
std::vector<T> do_eval(const std::vector<T> & x, const std::vector<T>& p)
{
T n01=std::abs(p[0]);
T beta1=p[1];
T rc1=p[2];
T n02=std::abs(p[3]);
T beta2=p[4];
T rc2=p[5];
std::vector<T> result(x.size()-1);
for(int i=1;i<x.size();++i)
{
T xi=(x[i]+x[i-1])/2;
T yi=0;
yi=n01*pow(1+xi*xi/rc1/rc1,-3./2.*beta1)+n02*pow(1+xi*xi/rc2/rc2,-3./2.*beta2);
result[i-1]=yi;
}
return result;
}
bool do_meets_constraint(const std::vector<T>& p)const
{
if(p.size()!=6)
{
std::cerr<<p.size()<<std::endl;
std::cerr<<this->get_num_params()<<std::endl;
assert(0);
}
T rc1=p.at(2);
T rc2=p.at(5);
if(rc2>rc1)
{
return true;
}
else
{
std::cerr<<rc2<<"\t"<<rc1<<std::endl;
std::cerr<<"***"<<std::endl;
return false;
}
}
};
}
#endif
|
