shap.models.TransformersPipeline

class shap.models.TransformersPipeline(pipeline, rescale_to_logits=False)[源代码]

这封装了一个 transformers 管道对象,以便于解释。

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

__init__(pipeline, rescale_to_logits=False)[源代码]

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

方法

__init__(pipeline[, rescale_to_logits])

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

load(in_file[, instantiate])

这是为了被子类重写并通过 super 调用。

save(out_file)

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

classmethod load(in_file, instantiate=True)

这是为了被子类重写并通过 super 调用。

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

save(out_file)

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