shap.models.TransformersPipeline
- class shap.models.TransformersPipeline(pipeline, rescale_to_logits=False)[源代码]
这封装了一个 transformers 管道对象,以便于解释。
默认情况下,transformers 管道对象输出的是字典列表,而不是 SHAP 所期望的标准张量。此类包装了管道,使其输出良好的张量格式。
方法
__init__
(pipeline[, rescale_to_logits])通过包装给定的管道对象来构建一个新模型。
load
(in_file[, instantiate])这是为了被子类重写并通过 super 调用。
save
(out_file)将模型保存到给定的文件流中。
- classmethod load(in_file, instantiate=True)
这是为了被子类重写并通过 super 调用。
当未被实例化时,我们返回构造函数参数值。由于 Serializable 类没有构造函数参数,我们只返回一个空字典。
- save(out_file)
将模型保存到给定的文件流中。