Changeset 1013 for library/bdm/estim/mixtures.cpp

Timestamp:

05/27/10 23:07:49 (14 years ago)

Author:

smidl

Message:

Flatten has an extra argument

Files:

• library/bdm/estim/mixtures.cpp (modified) (5 diffs)

library/bdm/estim/mixtures.cpp

@@ -9 +9 @@
         Coms.set_size ( c );
         weights.set_parameters ( ones ( c ) ); //assume at least one observation in each comp.
+        multiBM weights0(weights);
         //est will be done at the end
         //
@@ -18 +19 @@
         Coms ( 0 )->bayes_batch ( Data );
         // Flatten it to its original shape
-        Coms ( 0 )->flatten ( Com0 );
+        shared_ptr<BMEF> SharpCom((BMEF*)Coms(0)->_copy());
+        Coms ( 0 )->flatten ( Com0 ); 
 
         //Copy it to the rest
@@ -32 +34 @@
                 } else { // pick at random
                         int ind = (int) floor ( ndat * UniRNG.sample() );
-                        Coms ( i )->bayes ( Data.get_col ( ind ), empty_vec );
+                        Coms ( i )->bayes_weighted ( Data.get_col ( ind ), empty_vec, ndat/Coms.length() );
                 }
-                //flatten back to oringinal
-                Coms ( i )->flatten ( Com0 );
+                //sharpen to the sharp component
+                //Coms ( i )->flatten ( SharpCom.get(), 1.0/Coms.length() );
         }
-
 }
 
-double MixEF::bayes_batch ( const mat &data , const mat &cond, const vec &wData ) {
+double MixEF::bayes_batch_weighted ( const mat &data , const mat &cond, const vec &wData ) {
         int ndat = data.cols();
         int t, i, niter;
@@ -133 +134 @@
 };
 
-void MixEF::bayes ( const mat &data, const vec &cond = empty_vec ) {
-        this->bayes_batch ( data, cond, ones ( data.cols() ) );
-};
-
-
 double MixEF::logpred ( const vec &yt, const vec &cond =empty_vec) const {
 
@@ -162 +158 @@
 }
 
-void MixEF::flatten ( const BMEF* M2 ) {
+void MixEF::flatten ( const BMEF* M2, double weight=1.0 ) {
         const MixEF* Mix2 = dynamic_cast<const MixEF*> ( M2 );
         bdm_assert_debug ( Mix2->Coms.length() == Coms.length(), "Different no of coms" );
         //Flatten each component
         for ( int i = 0; i < Coms.length(); i++ ) {
-                Coms ( i )->flatten ( Mix2->Coms ( i ) );
+                Coms ( i )->flatten ( Mix2->Coms ( i ) , weight);
         }
         //Flatten weights = make them equal!!