normalize_labels.hpp File Reference
Include dependency graph for normalize_labels.hpp:
This graph shows which files directly or indirectly include this file:

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
eT
,
typename
RowType
>
void NormalizeLabels (const RowType &labelsIn, arma::Row< size_t > &labels, arma::Col< eT > &mapping)
 Given a set of labels of a particular datatype, convert them to unsigned labels in the range [0, n) where n is the number of different labels. More...

 
template
<
typename
eT
>
void RevertLabels (const arma::Row< size_t > &labels, const arma::Col< eT > &mapping, arma::Row< eT > &labelsOut)
 Given a set of labels that have been mapped to the range [0, n), map them back to the original labels given by the 'mapping' vector. More...

 

Detailed Description

Author
Ryan Curtin

Often labels are not given as {0, 1, 2, ...} but instead {1, 2, ...} or even {-1, 1} or otherwise. The purpose of this function is to normalize labels to {0, 1, 2, ...} and provide a mapping back to those labels.

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 normalize_labels.hpp.