Changeset 233 for trunk/src/WeNNI.cc

Timestamp:

Feb 21, 2005, 3:52:30 PM (18 years ago)

Author:

Peter

Message:

function added

File:

• trunk/src/WeNNI.cc (modified) (3 diffs)

trunk/src/WeNNI.cc

@@ -15 +15 @@
   WeNNI::WeNNI(const gslapi::matrix& matrix,const gslapi::matrix& flag,
                const u_int neighbours)
-    : NNI(matrix,flag,neighbours)
+    : NNI(matrix,flag,neighbours), imputed_data_raw_(matrix)
   {
     //estimate();
@@ -34 +34 @@
       sort(distance.begin(),distance.end(),
                 pair_value_compare<u_int,double>());
+      bool row_imputed=true;
       for (unsigned int j=0; j<data_.columns(); j++) {
         vector<u_int> knn=nearest_neighbours(j,distance);
@@ -47 +48 @@
         }
         // No impute if no contributions from neighbours.
-        if (norm)
+        if (norm){
+          imputed_data_raw_(i,j) = new_value/norm;
           imputed_data_(i,j)=
-            weight_(i,j)*data_(i,j) + (1-weight_(i,j)) * new_value/norm;
+            weight_(i,j)*data_(i,j) + (1-weight_(i,j))* imputed_data_raw_(i,j);
+        }
         else
-          not_imputed_.push_back(i);
+          row_imputed=false;
       }
+      if (!row_imputed)
+        not_imputed_.push_back(i);
     }
     return not_imputed_.size();