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 int totalElems = static_cast<int>(input.n_elem);
  const T *inPtr = input.memptr();
  T *outPtr = output.memptr();

  #pragma omp parallel for
  for (int i = 0; i < totalElems; ++i)
  {
    if (inPtr[i] > threshold)
      outPtr[i] = 1;
    else
      outPtr[i] = 0;
  }
}

template<typename T>
void Binarize(const arma::Mat<T>& input,
              arma::Mat<T>& output,
              const double threshold,
              const size_t dimension)
{
  output = input;
  const int totalCols = static_cast<int>(input.n_cols);

  #pragma omp parallel for
  for (int i = 0; i < totalCols; ++i)
  {
    if (input(dimension, i) > threshold)
      output(dimension, i) = 1;
    else
      output(dimension, i) = 0;
  }
}

} // namespace data
} // namespace mlpack

#endif