在另一個代碼中使用scikit SVM生成模型

Question

我已經訓練了一個名爲main.py的python腳本中的LinearSVC，用於訓練圖像分類算法。模型看起來像這樣。

lin_clf = svm.LinearSVC() 
lin_clf.fit(feature,g) 

我需要使用這個訓練好的模型來預測另一個代碼中的圖像類。我如何將基因處理模型lin_clf導出到其他代碼。

預先感謝您。

Answer 1

我知道你的「其他代碼」是另一個python腳本。 在這種情況下，您肯定可以使用pickle或shelve模塊將lin_clf寫入main.py中的磁盤，並在將使用該模型的腳本中從磁盤中讀取它。

這裏是示出如何使用寫lin_clf對象到磁盤的示例擱置：

import shelve 
a = shelve.open("output") 
a['lin_clf'] = lin_clf 
a.close()