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

我使用scikit-learn创建了一个SVM分类器;它基本上是读取一个描述,然后根据该描述返回一个类别。

目前,我使用FLASK框架创建了一个API,以便将我的调用发送到一个集中式的Linux服务器。然而,我每次使用它时都需要先pickle然后再unpickle,我希望创建一个持久加载的模型以加快我的分类时间。

  1. 是否有人能推荐一些资源,或者
  2. 指导我如何实现这一点?

回答:

import picklemodel = pickle.load(path)def predict(val):    return model.predict(val)

在函数外部声明文件可以使它们在API运行期间保持在内存中。从你的Flask脚本中调用predict函数。

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注