binarize.hpp

Go to the documentation of this file.

 
#ifndef MLPACK_CORE_DATA_BINARIZE_HPP
#define MLPACK_CORE_DATA_BINARIZE_HPP
 
#include <mlpack/prereqs.hpp>
 
namespace mlpack {
namespace data {
 
template<typename T>
void Binarize(const arma::Mat<T>& input,
              arma::Mat<T>& output,
              const double threshold)
{
  output.copy_size(input);
 
  const T *inPtr = input.memptr();
  T *outPtr = output.memptr();
 
  #pragma omp parallel for
  for (omp_size_t i = 0; i < (omp_size_t) input.n_elem; ++i)
    outPtr[i] = inPtr[i] > threshold;
}
 
template<typename T>
void Binarize(const arma::Mat<T>& input,
              arma::Mat<T>& output,
              const double threshold,
              const size_t dimension)
{
  output = input;
 
  #pragma omp parallel for
  for (omp_size_t i = 0; i < (omp_size_t) input.n_cols; ++i)
    output(dimension, i) = input(dimension, i) > threshold;
}
 
} // namespace data
} // namespace mlpack
 
#endif