如何在内存中加载 scikit-learn 机器学习模块?

How to load a scikit-learn machine learning module in memory?

我使用 scikit-learn 创建了一个 SVM 分类器;基本上它会引入一个描述,然后 returns 一个基于该描述的类别。

目前我已经使用 FLASK 框架创建了一个 API,以便将我的呼叫发送到一个集中的 Linux 框。但是,每次使用它时,我一直在 pickling 和 unpickling,但我想创建一个持久加载模型以加快我的分类时间。

  1. 是否有人们可以推荐的资源或
  2. 指导我如何完成此操作?
import pickle

model = pickle.load(path)

def predict(val):
    return model.predict(val)

在函数外部声明文件会将它们保存在内存中,直到 api 为 运行。 从您的 Flask 脚本调用预测函数。