source: trunk/yat/classifier/KNN_ReciprocalRank.cc @ 2881

// $Id: KNN_ReciprocalRank.cc 2881 2012-11-18 01:28:05Z peter $

/*
  Copyright (C) 2008 Jari Häkkinen, Peter Johansson, Markus Ringnér

  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 <config.h>

#include "KNN_ReciprocalRank.h"
#include "Target.h"

#include "yat/utility/VectorBase.h"
#include "yat/utility/VectorMutable.h"

#include <cmath>
#include <vector>

namespace theplu {
namespace yat {
namespace classifier {

  void KNN_ReciprocalRank::operator()(const utility::VectorBase& distance,
                                      const std::vector<size_t>& k_sorted, 
                                      const Target& target, 
                                      utility::VectorMutable& prediction) const
  {           
    for(size_t j=0;j<k_sorted.size();j++) 
      if(!std::isinf(distance(k_sorted[j])))
         prediction(target(k_sorted[j]))+=1.0/(j+1);                
  }

}}}