2016-11-27 122 views
0

我正在研究與人臉檢測情感有關的項目。我使用SVM首先訓練圖像,並使用這些訓練數據來預測新圖像。但在測試新圖像之前,我必須訓練所有的時間。是否有任何方法來存儲訓練數據並每次使用它來測試新圖像?在SVM中只訓練一次python

clf = SVC(kernel='linear', probability=True, tol=1e-3) 
def train(): 
    train_data,train_label = trainfiles() 
    data = np.array(train_data) 
    label = np.array(train_label) 
    clf.fit(data,label) 
+0

當您嘗試一個新的形象,你需要輸入新的訓練數據嗎? 或者您是否需要一種方法來保存模型並在測試數據上使用它? –

+0

@MohamedALANI只需保存一次模型,然後使用它來測試新圖像。 –

回答

1

量詞只是對象,你可以轉儲像像任何其他:

保存分類

from sklearn.externals import joblib 
joblib.dump(clf, 'filename.pkl') 

再次裝入

clf = joblib.load('filename.pkl') 
相關問題