【发布时间】:2017-08-19 14:07:14
【问题描述】:
我使用 SVM 分类器构建了情绪分析器。我用概率=真训练模型,它可以给我概率。但是当我腌制模型并稍后再次加载时,概率不再起作用。
型号:
from sklearn.svm import SVC, LinearSVC
pipeline_svm = Pipeline([
('bow', CountVectorizer()),
('tfidf', TfidfTransformer()),
('classifier', SVC(probability=True)),])
# pipeline parameters to automatically explore and tune
param_svm = [
{'classifier__C': [1, 10, 100, 1000], 'classifier__kernel': ['linear']},
{'classifier__C': [1, 10, 100, 1000], 'classifier__gamma': [0.001, 0.0001], 'classifier__kernel': ['rbf']},
]
grid_svm = GridSearchCV(
pipeline_svm,
param_grid=param_svm,
refit=True,
n_jobs=-1,
scoring='accuracy',
cv=StratifiedKFold(label_train, n_folds=5),)
svm_detector_reloaded = cPickle.load(open('svm_sentiment_analyzer.pkl', 'rb'))
print(svm_detector_reloaded.predict([""""Today is awesome day"""])[0])
给我:
AttributeError:predict_proba 在probability=False 时不可用
【问题讨论】:
-
能否把原来保存对象的代码显示到
''svm_sentiment_analyzer.pkl''? -
在收到
AttributeError时,您是否尝试拨打predict_proba而不是predict?否则这有点令人费解
标签: python scikit-learn svc