source: trunk/src/kNNI.cc @ 172

// $Id: kNNI.cc 172 2004-09-28 09:53:45Z jari $

#include "kNNI.h"

#include <algorithm>
#include <cmath>
#include <fstream>
#include <vector>

#include "stl_utility.h"

namespace theplu {
namespace cpptools {

  kNNI::kNNI(gslapi::matrix& matrix,const gslapi::matrix& flag,
             const u_int neighbours)
    : NNI(matrix,flag,neighbours)
  {
    for (unsigned int i=0; i<weight_.rows(); i++)
      for (unsigned int j=0; j<weight_.columns(); j++)
        if (!weight_(i,j)) {
          mv_rows_.push_back(i);
          break;
        }
    estimate();
  }



// \hat{x_{j}}=\frac{ \sum_{1,k} \frac{x_i}{d_{ij}} }
//                  { \sum_{1,k} \frac{1  }{d_{ij}} },
  void kNNI::estimate(void)
  {
    using namespace std;
    for (unsigned int i=0; i<mv_rows_.size(); i++) {
      // Jari, avoid copying in next line
      vector<pair<u_int,double> > distance=calculate_distances(mv_rows_[i]);
      sort(distance.begin(),distance.end(),
                pair_value_compare<u_int,double>());
      for (unsigned int j=0; j<data_.columns(); j++)
        if (!weight_(mv_rows_[i],j)) {
          vector<u_int> knn=nearest_neighbours(j,distance);
          double new_value=0.0;
          double norm=0.0;
          for (vector<u_int>::const_iterator k=knn.begin(); k!=knn.end(); k++) {
            new_value+=data_(distance[*k].first,j)/(distance[*k].second);
            norm+=1.0/(distance[*k].second);
          }
          // No impute if no contributions from neighbours.
          if (norm)
            imputed_data_(mv_rows_[i],j)=new_value/norm;
        }
    }
  }


}} // of namespace cpptools and namespace theplu