source: trunk/src/kNNI.h @ 178

// $Id: kNNI.h 178 2004-10-01 09:24:13Z jari $

#ifndef _theplu_cpptools_knni_
#define _theplu_cpptools_knni_ 

#include "NNI.h"

#include <iostream>
#include <vector>

namespace theplu {
namespace cpptools {

  ///
  /// kNNI is the binary weight implementation of NNI. This follows
  /// the work done by Troyanskaya et al. cited in the NNI document
  /// referred to in the NNI class documentation.
  ///
  /// This is a special case of the WeNNI, but is maintained since it
  /// is faster than the more general WeNNI.
  ///
  /// @see NNI and WeNNI
  ///
  class kNNI : public NNI
  {
  public:
    /// Perform kNNI on data in \a matrix with binary uncertainty
    /// weights in \a weight using \a neighbours for the new impute
    /// value.
    ///
    /// @note The algorithm is initiated within the constructor and
    /// the result is fetched with the NNI::imputed_data method.
    kNNI(const gslapi::matrix& matrix,const gslapi::matrix& weight,
         const u_int neighbours);

  private:
    void estimate(void);

    vector<u_int> mv_rows_; // index to rows that have values to estimate
  };

}} // of namespace cpptools and namespace theplu

#endif