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?