Source code for immuneML.environment.Label

import logging

[docs] class Label: def __init__(self, name: str, values: list, auxiliary_label_names: list = None, positive_class=None): = name self._values = values self.auxiliary_label_names = auxiliary_label_names self._positive_class = positive_class def __str__(self): return f"label {} ({', '.join([str(val) for val in self.values])})" @property def positive_class(self): if self._positive_class is not None: return self._positive_class else: assert self._values is not None, f"Label: cannot access positive class for label {} " \ f"when neither 'positive_class' nor 'values' are set." positive_class = sorted(self._values)[0]"Label: No positive class was set for label {}. " f"Assuming default positive class '{positive_class}'.") return positive_class @property def values(self): ''' Make sure the positive class is listed last This is needed for compatibility with `~immuneML.ml_methods.util.Util.binarize_label_classes` ''' if self._positive_class is not None: negative_classes = sorted(self._values) negative_classes.remove(self._positive_class) return negative_classes + [self._positive_class] else: return sorted(self._values)
[docs] def get_binary_negative_class(self): assert len(self._values) == 2, f"Label: binary negative class was requested for label {} but this label contains {len(self._values)} classes: {self.values}" return [val for val in self._values if val != self.positive_class][0]
[docs] def get_desc_for_storage(self): return {key.lstrip("_"): value for key, value in vars(self).items()}