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]