src.model.deeplearn.handle.dl_label_formatter

Classes

DLLabelFormatter()

Class to handle the labels for deep learning models.

class src.model.deeplearn.handle.dl_label_formatter.DLLabelFormatter

Class to handle the labels for deep learning models.

__init__()
handle_labels_format(mh, y)

Handles the format in which labels must be given to the model.

For instance, if categorical cross entropy is used, labels must be given using one-hot-encoding. However, if sparse categorical cross entropy is used, labels must be given as an integer.

Parameters:
  • mh (DLModelHandler) – The model handler whose labels must be formatted.

  • y – The labels to be formatted.

Returns:

The labels prepared for the model.