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

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.