如何在Streamlight中缓存绘图？

Question

我在Streamlight中构建了一个仪表板，您可以在其中选择客户ID并显示形状图（瀑布图和强制图），以解释该客户的信用违约预测。

我还想显示一个包含整个列车数据集的SHAP摘要图。后者不会在每次进行新预测时更改，并且需要花费大量时间来绘制，因此我希望缓存它。我想最好的方法是使用st.cache，但我一直没能做到。

下面是我在main.py中尝试过但没有成功的代码：我首先定义了我想要缓存输出的函数（图），然后我在st.pyplot.中执行输出>装饰器，但是一旦我添加它并重新运行应用程序，函数summary_plot_all无限期运行

在：

@st.cache    
def summary_plot_all():
    fig, axes = plt.subplots(nrows=1, ncols=1)
    shap.summary_plot(shapvs[1], prep_train.iloc[:, :-1].values, 
    prep_train.columns, max_display=50)
    return fig
    
st.pyplot(summary_plot_all())

OUT（在流光应用程序中显示）

正在运行摘要\绘图\所有（）。

有人知道什么是错的或更好的方法缓存一个情节在流？

version of packages:
streamlit==0.84.1, 
matplotlib==3.4.2, 
shap==0.39.0

Answer 1

试试看

import matplotlib

@st.cache(hash_funcs={matplotlib.figure.Figure: lambda _: None})
def summary_plot_all():
    fig, axes = plt.subplots(nrows=1, ncols=1)
    shap.summary_plot(shapvs[1], prep_train.iloc[:, :-1].values, 
    prep_train.columns, max_display=50)
    return fig

检查此streamlightgithub问题