Source code for draugr.torch_utilities.datasets.supervised.categorical_dataset

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

__author__ = "Christian Heider Nielsen"
__doc__ = r"""

           Created on 01/03/2020
           """

from abc import abstractmethod
from typing import Sequence

from draugr.torch_utilities.datasets.supervised.supervised_dataset import (
    SupervisedDataset,
)

__all__ = ["CategoricalDataset"]

from warg import OrderedSet


[docs]class CategoricalDataset(SupervisedDataset): """ Categorical Dataset for discrete learning problems.""" @property @abstractmethod def categories(self) -> OrderedSet[str]: """description""" raise NotImplementedError
[docs] def idx_to_str(self, idx: Sequence[int]) -> Sequence[str]: """description""" return self.categories[idx]
[docs] def str_to_idx(self, s: str) -> int: """description""" return self.categories.index(s)