shap.models.TransformersPipeline

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

这封装了一个 transformers pipeline 对象以便于解释。

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

__init__(pipeline, rescale_to_logits=False)

通过封装给定的 pipeline 对象构建新模型。

方法

__init__(pipeline[, rescale_to_logits])

通过封装给定的 pipeline 对象构建新模型。

load(in_file[, instantiate])

这旨在被子类重写并使用 super 调用。

save(out_file)

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

classmethod load(in_file, instantiate=True)

这旨在被子类重写并使用 super 调用。

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

save(out_file)

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