cap_k.hpp

Go to the documentation of this file.

#ifndef KNNCOLLE_CAP_K_HPP
#define KNNCOLLE_CAP_K_HPP
 
#include "sanisizer/sanisizer.hpp"
 
namespace knncolle {
 
template<typename Index_>
int cap_k(int k, Index_ num_observations) {
    if (sanisizer::is_less_than(k, num_observations)) {
        return k;
    } else if (num_observations) {
        return num_observations - 1;
    } else {
        return 0;
    }
}
 
template<typename Index_>
int cap_k_query(int k, Index_ num_observations) {
    return sanisizer::min(k, num_observations);
}
 
}
 
#endif