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)
将模型保存到给定的文件流。