LeapText

code_loader.contract.visualizer_classes.LeapText

Used to visualize Text. Optionally, together with a heatmap that signifies words importance

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

@dataclass
class LeapText:
    data: List[str]
    type: LeapDataType = LeapDataType.Text
    heatmap: Optional[List[float]] = None
Args

data

a list of strings that compose the text (length N)

heatmap

(optional) - a lists of floats that provides word importance (length N)

Examples

Basic Usage

import numpy as np
from code_loader.contract.visualizer_classes import LeapText
from code_loader.contract.enums import LeapDataType
...

@tensorleap_custom_visualizer(name="input_visualizer", visualizer_type=LeapDataType.Text)
def input_visualizer(input_ids: np.ndarray) -> LeapText:
    input_ids = np.squeeze(input_ids)
    text = decode_token_ids(input_ids)
    return LeapText(text)

Last updated

Was this helpful?