簡體   English   中英

如何在內存中加載scikit學習機器學習模塊?

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

我使用scikit-learn創建了一個SVM分類器; 基本上,它提取一個描述,然后根據該描述返回一個類別。

目前,我已經使用FLASK框架創建了一個API,以便將調用發送到集中式Linux機器。 但是,我一直在對其進行腌制然后每次對其進行腌制,但我想創建一個持久的加載模型以加快分類時間。

  1. 人們是否可以推薦資源?
  2. 指導我如何做到這一點?
import pickle

model = pickle.load(path)

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

在函數外部聲明文件會將其保留在內存中,直到api運行為止。 從燒瓶腳本中調用預測函數。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM