2012-07-11 73 views
3

我想在其他计算机上使用此分类器,而无需再次进行训练。 我曾经用cPickle从scikit保存一些分类器。 与LIBSVM一样,它给了我一个“ValueError:包含指针的ctypes对象不能被腌制”。如何保存LibSVM python对象实例?

我正在使用LibSVM 3.1和Python 2.7.3。

感谢

from libsvm.svm import * 
from libsvm.svmutil import * 
import cPickle 

x = [[1, 0, 1], [-1, 0, -1]] 
y = [1, -1] 
prob = svm_problem(y, x) 
param = svm_parameter() 
param.kernel_type = LINEAR 
param.C = 10 
m = svm_train(prob, param) 
labels_pred, acc, probs = svm_predict([-1, 1], [[1, 1, 1], [0, 0, 1]], m) 
print labels_pred, acc, probs 

import ipdb; ipdb.set_trace() 

filename='libsvm-classif.pkl' 

fid = open(filename, 'wb') 
cPickle.dump(m, fid) 
fid.close() 

fid = open(filename, 'rb') 
m = cPickle.load(fid) 
labels_pred, acc, probs = svm_predict([-1, 1], [[1, 1, 1], [0, 0, 1]], m) 

print labels_pred, acc, probs 

回答

9

只要使用LIBSVM的加载和保存功能

svm_save_model('libsvm.model', m) 
m = svm_load_model('libsvm.model') 

这是包含在LIBSVM包的蟒蛇目录中的README文件。它似乎比网站有更好的功能描述。

+0

谢谢。工作很好。 =) – 2012-08-13 14:21:55