ferret.GradientExplainer

class ferret.GradientExplainer(model, tokenizer, model_helper: str | None = None, multiply_by_inputs: bool = True, **kwargs)

__init__(model, tokenizer, model_helper: str | None = None, multiply_by_inputs: bool = True, **kwargs)

Methods

__init__(model, tokenizer[, model_helper, ...])
compute_feature_importance(text[, target, ...])
get_input_embeds(text)
get_tokens(text)

Attributes

NAME
device
model
tokenizer