one_hot_encoding.hpp File Reference
Include dependency graph for one_hot_encoding.hpp:

Go to the source code of this file.

Namespaces

 mlpack
 
Linear algebra utility functions, generally performed on matrices or vectors.
 
 mlpack::data
 
Functions to load and save matrices and models.
 

Functions

template
<
typename
RowType
,
typename
MatType
>
void OneHotEncoding (const RowType &labelsIn, MatType &output)
 Given a set of labels of a particular datatype, convert them to binary vector. More...

 
template
<
typename
eT
>
void OneHotEncoding (const arma::Mat< eT > &input, const arma::Col< size_t > &indices, arma::Mat< eT > &output)
 Overloaded function for the above function, which takes a matrix as input and also a vector of indices to encode and outputs a matrix. More...

 
template
<
typename
eT
>
void OneHotEncoding (const arma::Mat< eT > &input, arma::Mat< eT > &output, const data::DatasetInfo &datasetInfo)
 Overloaded function for the above function, which takes a matrix as input and also a DatasetInfo object and outputs a matrix. More...

 

Detailed Description

Author
Jeffin Sam

One hot encoding functions. The purpose of this function is to convert categorical variables to binary vectors.

mlpack is free software; you may redistribute it and/or modify it under the terms of the 3-clause BSD license. You should have received a copy of the 3-clause BSD license along with mlpack. If not, see http://www.opensource.org/licenses/BSD-3-Clause for more information.

Definition in file one_hot_encoding.hpp.