source: trunk/src/kNNI.h @ 228

// $Id: kNNI.h 228 2005-02-01 14:06:51Z peter $

#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:
    ///
    /// Constructor
    ///
    kNNI(const gslapi::matrix& matrix,const gslapi::matrix& weight,
         const u_int neighbours);

    ///
    /// Perform kNNI on data in \a matrix with binary uncertainty
    /// weights in \a weight using \a neighbours for the new impute
    /// value.
    ///
    size_t estimate(void);

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

}} // of namespace cpptools and namespace theplu

#endif