NNI.h

#ifndef _theplu_yat_utility_nni_
#define _theplu_yat_utility_nni_ 

// $Id: NNI.h 2119 2009-12-12 23:11:43Z peter $

/*
  Copyright (C) 2004 Jari Häkkinen
  Copyright (C) 2005, 2006, 2007, 2008 Jari Häkkinen, Peter Johansson
  Copyright (C) 2009 Jari Häkkinen

  This file is part of the yat library, http://dev.thep.lu.se/yat

  The yat library is free software; you can redistribute it and/or
  modify it under the terms of the GNU General Public License as
  published by the Free Software Foundation; either version 3 of the
  License, or (at your option) any later version.

  The yat library is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with yat. If not, see <http://www.gnu.org/licenses/>.
*/

#include "Matrix.h"

#include <utility>
#include <vector>

namespace theplu {
namespace yat {
namespace utility {

  class NNI
  {
  public:

    NNI(const utility::Matrix& matrix,const utility::Matrix& weight,
        const unsigned int neighbours);

    virtual ~NNI(void) {};

    virtual unsigned int estimate(void)=0;

    const utility::Matrix& imputed_data(void) const;

    const std::vector<size_t>& not_imputed(void) const;

  protected:
    std::vector<std::pair<size_t,double> > 
    calculate_distances(const size_t) const;
    std::vector<size_t> 
    nearest_neighbours(const size_t,
                       const std::vector<std::pair<size_t,double> >&) const;
    const utility::Matrix& data_;

    utility::Matrix imputed_data_;

    unsigned int neighbours_;

    std::vector<size_t> not_imputed_;

    const utility::Matrix& weight_;
  };

}}} // of namespace utility, yat, and theplu

#endif