【大模型---streamlit中的装饰器st.cache_resource()】

文章介绍了Streamlit中的@st.cache_resource装饰器,用于缓存数据库连接等不可序列化资源,提高应用性能。通过ttl设置缓存过期时间,展示了其在减少重复创建资源和提升效率方面的应用。
摘要由CSDN通过智能技术生成

@st.cache_resource 是 Streamlit 提供的一个装饰器,用于缓存返回全局资源(如数据库连接、机器学习模型等)的函数。这些缓存的对象可以在所有用户、会话和重新运行中全局可用。这个装饰器特别适合于那些本质上不可序列化的类型,例如数据库连接、文件句柄或线程等,但也可以用于可序列化的对象。缓存的对象必须是线程安全的,因为它们可能会被多个线程同时访问。如果线程安全成为问题,可以考虑使用 st.session_state 来存储每个会话的资源。

@st.cache_resource 装饰器的使用示例如下:

import streamlit as st
import pandas as pd
from sqlalchemy import create_engine

@st.cache_resource(ttl=10800) # 3小时过期
def pd_mysql():
    host='localhost'
    port=3306
    username='root'
    password='123456'
    db='mydb'

    db_url = f'mysql+mysqlconnector://{username}:{password}@{host}:{port}/{db}'
    engine = create_engine(db_url)
    return engine

engine = pd_mysql() # 创建引擎

在这个例子中,@st.cache_resource 用于缓存数据库连接引擎,避免了每次运行时都重新创建连接,从而提高了应用的性能。ttl 参数指定了缓存的过期时间,例如 ttl=10800 表示缓存的对象在3小时后过期1。

此外,@st.cache_resource 还提供了一些其他参数,如 max_entries 用于限制缓存中的最大条目数,show_spinner 用于控制是否显示加载指示器,validate 用于提供一个验证函数来检查缓存数据的有效性,以及 hash_funcs 用于自定义哈希函数。

总的来说,@st.cache_resource 是一个强大的工具,用于优化 Streamlit 应用的性能,特别是在处理全局资源时。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
`st.cache_resource` 是 `streamlit` 库的一个函数,可以帮助你缓存应用程序使用的外部数据资源,例如文件、URL 等。使用 `st.cache_resource` 函数,你可以将外部资源的读取和处理结果缓存到本地磁盘,从而加速应用程序的运行。 `st.cache_resource` 函数的定义如下: ```python def cache_resource(func, resource_func=None, suppress_st_warning=False): ``` 其,`func` 是要缓存的函数,`resource_func` 是读取外部资源的函数,`suppress_st_warning` 用于控制是否显示警告信息。 下面是一个简单的示例,展示如何使用 `st.cache_resource` 函数缓存一个外部数据资源: ```python import streamlit as st @st.cache_resource def get_data(): # 读取外部数据资源 data = pd.read_csv("data.csv") # 对数据进行处理 data = data.groupby("category")["value"].sum() return data # 在页面上显示处理后的数据 data = get_data() st.write(data) ``` 在这个示例,我们使用 `st.cache_resource` 装饰器将 `get_data` 函数标记为要缓存的函数。在 `get_data` 函数,我们首先读取了一个外部数据资源(例如一个 CSV 文件),然后对数据进行了处理。由于 `get_data` 函数被标记为要缓存的函数,`streamlit` 库会自动将函数的输入和输出缓存到本地磁盘,从而加速应用程序的运行。 需要注意的是,`st.cache_resource` 函数只适用于读取外部数据资源的函数,如果要缓存其他类型的函数,应该使用 `st.cache` 函数。 总之,`st.cache_resource` 函数是 `streamlit` 库一个非常有用的函数,可以帮助你缓存应用程序使用的外部数据资源,加速应用程序的运行。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值