lfxai.utils.feature_attribution module

generate_masks(attr: ndarray, mask_size: int) → Tensor

Generates mask for images with feature importance scores :param attr: feature importance scores :param mask_size: number of pixels masked

Returns

Mask hiding most important pixels

generate_tseries_masks(attr: ndarray, mask_size: int) → Tensor

Generates mask for time series with feature importance scores :param attr: feature importance scores :param mask_size: number of time steps masked

Returns

Mask hiding most important time steps