首页 > 解决方案 > 如何与使用 python pickle 模块保存的 ML 模型进行交互?

问题描述

我使用 pickle 模块保存了一个使用 pandas、numpy 和 statsmodels 库编写的 ML 模型。

然后我测试将它加载到一个新的 jupyter 笔记本中,虽然它似乎加载成功,但我不确定如何从笔记本中与之交互。这是我第一次尝试这样的事情,所以我不知道如何开始。

这是加载模型的代码。'Loaded'打印在输出中 - 没有其他错误再次出现。

# load your model
import pickle                         
with open('./model.pkl', 'rb') as f:
    model = pickle.load(f)
    print('Loaded')

标签: pythonjupyter-notebookpicklestatsmodels

解决方案


这是一个普通的模型,就好像你从来没有腌制过它一样。pickle 的神奇之处在于您可以保存(几乎)任何对象,稍后加载并毫无问题地使用它。


推荐阅读