rex_xai.prediction

Classes

Prediction

Functions

from_pytorch_tensor(tensor[, target, binary_threshold])

Module Contents

class rex_xai.prediction.Prediction(pred=None, conf=None, box=None, target=None, target_confidence=None)
classification: int | None = None
confidence: float | None = None
bounding_box: numpy.typing.NDArray | None = None
target: int | None = None
target_confidence: float | None = None
__repr__()
Return type:

str

get_class()
is_empty()
is_passing()
rex_xai.prediction.from_pytorch_tensor(tensor, target=None, binary_threshold=None)
Return type:

List[Prediction]