Changeset 1331 for applications/robust/robustlib.h

Timestamp:

04/14/11 19:24:43 (13 years ago)

Author:

sindj

Message:

Dodelan sampling sigma, zbyva doladit sampling alpha.JS

Files:

• applications/robust/robustlib.h (modified) (5 diffs)

applications/robust/robustlib.h

@@ -67 +67 @@
 class simplex
 {
+        
+
 public:
 
@@ -73 +75 @@
         double probability;
 
-        vector<multimap<double,double>> gamma_parameters;
-
-        double gamma_sum;
+        vector<multimap<double,double>> positive_gamma_parameters;
+
+        vector<multimap<double,double>> negative_gamma_parameters;
+
+        double positive_gamma_sum;
+
+        double negative_gamma_sum;
         
 
@@ -88 +94 @@
                 this->vertices.insert(vertex);
                 probability = 0;
+        }
+
+        void clear_gammas()
+        {
+                positive_gamma_parameters.clear();
+                negative_gamma_parameters.clear();
+                
+                
+                positive_gamma_sum = 0;
+                negative_gamma_sum = 0;
+        }
+
+        void insert_gamma(int order, double weight, double beta)
+        {
+                if(weight>=0)
+                {
+                        if(positive_gamma_parameters.size()<order+1)
+                        {
+                                multimap<double,double> map;
+                                positive_gamma_parameters.push_back(map);
+                        }
+
+                        positive_gamma_sum += weight;
+
+                        positive_gamma_parameters[order].insert(pair<double,double>(weight,beta));              
+                }
+                else
+                {
+                        if(negative_gamma_parameters.size()<order+1)
+                        {
+                                multimap<double,double> map;
+                                negative_gamma_parameters.push_back(map);
+                        }
+
+                        negative_gamma_sum -= weight;
+
+                        negative_gamma_parameters[order].insert(pair<double,double>(-weight,beta));
+                }
         }
 };
@@ -1742 +1786 @@
 
         mat sample_mat(int n)
-        {
-                
-                
+        {               
 
                 /// \TODO tady je to spatne, tady nesmi byt conditions.size(), viz RARX.bayes()
@@ -1796 +1838 @@
                                 // cout << &(*tri_ref) << endl;
 
-                                rnumber = randu();
-
+                                bool have_sigma = false;
                                 double sigma = 0;
-                                double sum_g = 0;
-                                for(int i = 0;i<(*s_ref)->gamma_parameters.size();i++)
-                                {
-                                        for(multimap<double,double>::iterator g_ref = (*s_ref)->gamma_parameters[i].begin();g_ref != (*s_ref)->gamma_parameters[i].end();g_ref++)
-                                        {
-                                                sum_g += (*g_ref).first/(*s_ref)->gamma_sum;
-
-                                                if(sum_g>rnumber)
-                                                {
-                                                        itpp::Gamma_RNG* gamma = new itpp::Gamma_RNG(conditions.size()-number_of_parameters,(*g_ref).second);
-                                                        sigma = (*gamma)();
+                                do
+                                {
+                                        rnumber = randu();
+                                        
+                                        double sum_g = 0;
+                                        for(int i = 0;i<(*s_ref)->positive_gamma_parameters.size();i++)
+                                        {
+                                                for(multimap<double,double>::iterator g_ref = (*s_ref)->positive_gamma_parameters[i].begin();g_ref != (*s_ref)->positive_gamma_parameters[i].end();g_ref++)
+                                                {
+                                                        sum_g += (*g_ref).first/(*s_ref)->positive_gamma_sum;
+
+                                                        if(sum_g>rnumber)
+                                                        {
+                                                                itpp::Gamma_RNG* gamma = new itpp::Gamma_RNG(conditions.size()-number_of_parameters,1/(*g_ref).second);
+                                                                sigma = 1/(*gamma)();
+                                                                break;
+                                                        }                                               
+                                                }
+
+                                                if(sigma!=0)
+                                                {
                                                         break;
-                                                }                                               
-                                        }
-
-                                        if(sigma!=0)
-                                        {
-                                                break;
-                                        }
-                                }
+                                                }
+                                        }
+
+                                        rnumber = randu();
+
+                                        double pg_sum = 0;
+                                        for(vector<multimap<double,double>>::iterator v_ref = (*s_ref)->positive_gamma_parameters.begin();v_ref!=(*s_ref)->positive_gamma_parameters.end();v_ref++)
+                                        {
+                                                for(multimap<double,double>::iterator pg_ref = (*v_ref).begin();pg_ref!=(*v_ref).end();pg_ref++)
+                                                {
+                                                        pg_sum += exp(-(*pg_ref).second/sigma)*pow((*pg_ref).second/sigma,(int)conditions.size()-number_of_parameters-1)*(*pg_ref).second/fact(conditions.size()-number_of_parameters-1)*(*pg_ref).first;
+                                                }                                       
+                                        }
+
+                                        double ng_sum = 0;
+                                        for(vector<multimap<double,double>>::iterator v_ref = (*s_ref)->negative_gamma_parameters.begin();v_ref!=(*s_ref)->negative_gamma_parameters.end();v_ref++)
+                                        {
+                                                for(multimap<double,double>::iterator ng_ref = (*v_ref).begin();ng_ref!=(*v_ref).end();ng_ref++)
+                                                {
+                                                        ng_sum += exp(-(*ng_ref).second/sigma)*pow((*ng_ref).second/sigma,(int)conditions.size()-number_of_parameters-1)*(*ng_ref).second/fact(conditions.size()-number_of_parameters-1)*(*ng_ref).first;
+                                                }                                       
+                                        }
+                                        
+                                        if((pg_sum-ng_sum)/pg_sum>rnumber)
+                                        {
+                                                have_sigma = true;
+                                        }
+                                }
+                                while(!have_sigma);
 
                                 int dimension = (*s_ref)->vertices.size()-1;