shap.models.TransformersPipeline

class shap.models.TransformersPipeline(pipeline, rescale_to_logits=False)

此包装器用于 transformers pipeline 对象，以便于解释。

默认情况下，transformers pipeline 对象输出字典列表，而不是 SHAP 期望的标准张量。此类别包装管道，使其输出良好的张量格式。

__init__(pipeline, rescale_to_logits=False)

通过包装给定的管道对象来构建新模型。

方法

__init__(pipeline[, rescale_to_logits])	通过包装给定的管道对象来构建新模型。
load(in_file[, instantiate])	这应由子类重写并用 super 调用。
save(out_file)	将模型保存到给定的文件流。

classmethod load(in_file: BinaryIO, instantiate: bool = True) → Model | dict[str, Any]

这应由子类重写并用 super 调用。

在未实例化时，我们返回构造函数参数值。由于 Serializable 类没有构造函数参数，我们只返回一个空字典。

save(out_file: BinaryIO) → None

将模型保存到给定的文件流。