source: trunk/src/WeNNI.h @ 246

// $Id: WeNNI.h 246 2005-02-23 19:17:31Z peter $

#ifndef _theplu_cpptools_wenni_
#define _theplu_cpptools_wenni_ 

#include "NNI.h"

#include <iostream>

#include "matrix.h"

namespace theplu {
namespace cpptools {

  using namespace std;

  ///
  /// WeNNI is a continuous weights generalization of the (binary
  /// weights) kNNI algorithm presented by Troyanskaya et al. A
  /// reference to this paper is found in the NNI document referred to
  /// in the NNI class documentation. The NNI document also describes
  /// WeNNI in depth.
  ///
  /// @see NNI and kNNI
  ///
  class WeNNI : public NNI
  {
  public:
    ///
    /// Constructor
    ///
    WeNNI(const gslapi::matrix& matrix,const gslapi::matrix& weight,
          const u_int neighbours);

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

    ///
    /// @return A const reference to imputed_data_raw.
    ///
    const gslapi::matrix& imputed_data_raw(void) const 
    { return imputed_data_raw_; }



  private:

    gslapi::matrix imputed_data_raw_;
  };

}} // of namespace cpptools and namespace theplu

#endif