one_hot_encoding.hpp

Go to the documentation of this file.

#ifndef MLPACK_CORE_DATA_ONE_HOT_ENCODING_HPP
#define MLPACK_CORE_DATA_ONE_HOT_ENCODING_HPP

#include <mlpack/prereqs.hpp>
#include <mlpack/core.hpp>

namespace mlpack {
namespace data {

template<typename RowType, typename MatType>
void OneHotEncoding(const RowType& labelsIn,
                    MatType& output);

template<typename eT>
void OneHotEncoding(const arma::Mat<eT>& input,
                    const arma::Col<size_t>& indices,
                    arma::Mat<eT>& output);

template<typename eT>
void OneHotEncoding(const arma::Mat<eT>& input,
                    arma::Mat<eT>& output,
                    const data::DatasetInfo& datasetInfo);

} // namespace data
} // namespace mlpack

// Include implementation.
#include "one_hot_encoding_impl.hpp"

#endif