normalize_labels.hpp

Go to the documentation of this file.

 
#ifndef MLPACK_CORE_DATA_NORMALIZE_LABELS_HPP
#define MLPACK_CORE_DATA_NORMALIZE_LABELS_HPP
 
#include <mlpack/prereqs.hpp>
 
namespace mlpack {
namespace data {
 
template<typename eT, typename RowType>
void NormalizeLabels(const RowType& labelsIn,
                     arma::Row<size_t>& labels,
                     arma::Col<eT>& mapping);
 
template<typename eT>
void RevertLabels(const arma::Row<size_t>& labels,
                  const arma::Col<eT>& mapping,
                  arma::Row<eT>& labelsOut);
 
} // namespace data
} // namespace mlpack
 
// Include implementation.
#include "normalize_labels_impl.hpp"
 
#endif