custom_imputation.hpp

Go to the documentation of this file.

 
#ifndef MLPACK_CORE_DATA_IMPUTE_STRATEGIES_CUSTOM_IMPUTATION_HPP
#define MLPACK_CORE_DATA_IMPUTE_STRATEGIES_CUSTOM_IMPUTATION_HPP
 
#include <mlpack/prereqs.hpp>
 
namespace mlpack {
namespace data {
template <typename T>
class CustomImputation
{
 public:
  CustomImputation(T customValue):
      customValue(std::move(customValue))
  {
    // nothing to initialize here
  }
 
  void Impute(arma::Mat<T>& input,
              const T& mappedValue,
              const size_t dimension,
              const bool columnMajor = true)
  {
    // replace the target value to custom value
    if (columnMajor)
    {
      for (size_t i = 0; i < input.n_cols; ++i)
      {
        if (input(dimension, i) == mappedValue ||
            std::isnan(input(dimension, i)))
        {
          input(dimension, i) = customValue;
        }
      }
    }
    else
    {
      for (size_t i = 0; i < input.n_rows; ++i)
      {
        if (input(i, dimension) == mappedValue ||
            std::isnan(input(i, dimension)))
        {
          input(i, dimension) = customValue;
        }
      }
    }
  }
 
 private:
  T customValue;
}; // class CustomImputation
 
} // namespace data
} // namespace mlpack
 
#endif