LeapHorizontalBar

code_loader.contract.visualizer_classes.LeapHorizontalBar

This object is used to visualize logits and histograms

import numpy.typing as npt
from code_loader.contract.enums import LeapDataType

@dataclass
class LeapHorizontalBar:
    body: npt.NDArray[np.float32]
    labels: List[str]
    gt: Optional[npt.NDArray[np.float32]] = None
    type: LeapDataType = LeapDataType.HorizontalBar
Args

data

np.ndarray float32 representation of the prediction confidences (length N) .

labels

a list of N strings representing the classes

gt

(optional) np.ndarray float32 representation of GT (length N).

Examples

Basic Usage

import numpy as np
from code_loader.contract.visualizer_classes import LeapHorizontalBar
from code_loader import leap_binder
from code_loader.contract.enums import LeapDataType
...


@tensorleap_custom_visualizer('label', visualizer_type=LeapDataType.HorizontalBar)
def animal_count_visualizer(num_of_cats: np.ndarray, num_of_dogs: np.ndarray) -> LeapHorizontalBar:
    body = [num_of_cats, num_of_dogs]
    labels = ['cats', 'dogs']
    return LeapHorizontalBar(body=body, labels=labels)

Last updated

Was this helpful?