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?