无法在 Streamlit 中显示 SHAP 文本可视化
Posted
技术标签:
【中文标题】无法在 Streamlit 中显示 SHAP 文本可视化【英文标题】：Cannot display SHAP text visualization in Streamlit 【发布时间】：2022-01-13 13:07:55 【问题描述】：我正在尝试构建我的 NLP 项目的仪表板。所以我使用 BERT 模型进行预测，使用 SHAP 包进行可视化，使用 Streamlit 来创建仪表板：
tokenizer = AutoTokenizer.from_pretrained(model_name_cla)
model = AutoModelForSequenceClassification.from_pretrained(model_name_cla)
labels = ['1- Tarife','2- Dateneingabe','3- Bestätigungsmail','4- Kundenbetreuung','5- Aufwand vom Vergleich bis Abschluss',
'6- After-sales Wechselprozess','7 - Werbung/VX Kommunikation','8 - Sonstiges','9 - Nicht auswertbar']
def f(x):
tv = torch.tensor([tokenizer.encode(v, padding='max_length', max_length=128, truncation=True) for v in x])
attention_mask = (tv!=0).type(torch.int64)
outputs = model(tv,attention_mask=attention_mask)[0].detach().cpu().numpy()
scores = (np.exp(outputs).T / np.exp(outputs).sum(-1)).T
val = sp.special.logit(scores)
return val
text = ['This is just a test']
# build an explainer using a token masker
explainer = shap.Explainer(f, tokenizer, output_names=labels)
shap_values = explainer(text, fixed_context=1)
shap.plots.text(shap_values)
代码在我的 jupyter 笔记本中运行良好，但是当我尝试将其作为 .py 文件执行时，没有任何反应。它既不显示任何内容，也不抛出错误。我的控制台只是在执行时返回这个：
>
如何在流光中显示我的图表？
【问题讨论】：
【参考方案1】：这可以用Streamlit Components 和最新的SHAP v0.36+（定义一个新的getjs 方法）可视化，以绘制JS SHAP plots
（一些像
summary_plot这样的图实际上是Matplotlib，可以用st.pyplot绘制）
import streamlit as st
import streamlit.components.v1 as components
def st_shap(plot, height=None):
shap_html = f"<head>shap.getjs()</head><body>plot.html()</body>"
components.html(shap_html, height=height)
st_shap(shap.plots.text(shap_values),400)
在 streamlit 此处 - Display SHAP diagrams with Streamlit 中查找有关可视化形状的更详细讨论
【讨论】：
我不知道为什么但我总是得到这个错误：AttributeError: 'NoneType' object has no attribute 'html' 你实际上得到了None html。与您在问题中添加的代码相同吗？
是的，完全一样。我还为此更新了 shap-
您能否检查一下您是否真的从数据中生成了任何可视化？？
当我在我的 jupyter 笔记本中执行代码 (shap.plots.text(shap_values)) 时，它会生成预期的情节。所以我想我正在生成一个......或者我错过了什么？以上是关于无法在 Streamlit 中显示 SHAP 文本可视化的主要内容，如果未能解决你的问题，请参考以下文章
错误安装 streamlit 它说“错误：无法为使用 PEP 517 且无法直接安装的 pyarrow 构建***”
通过docker运行时无法在浏览器中查看streamlit应用程序
使用 VConcat 在 Streamlit 中未显示 Altair 图表