python - 如何在 python 和 NLTK 中计算预测概率?
问题描述
我正在尝试通过使用LinearSVC
并OneVsRestClassifier
得到错误来计算 SVM 模型中的每个预测概率
AttributeError: 'LinearSVC' object has no attribute 'predict_proba'
试过的代码:
model = Pipeline([('vectorizer', CountVectorizer(ngram_range=(1,2))),
('tfidf', TfidfTransformer(use_idf=True)),
('clf', OneVsRestClassifier(LinearSVC(class_weight="balanced")))])
model.fit(X_train, y_train)
y_train.shape
pred = model.predict(X_test)
probas = model.predict_proba(X_test)
也试过:
from nltk.classify.scikitlearn import SklearnClassifier
from sklearn.svm import SVC
LinearSVC_classifier = SklearnClassifier(SVC(kernel='linear',probability=True))
prob_1 = LinearSVC_classifier.predict_proba(X_test)
但仍然出现错误AttributeError: 'SklearnClassifier' object has no attribute 'predict_proba'
请提出相同的建议。
解决方案
使用您的线性 SVM:
from sklearn.calibration import CalibratedClassifierCV
from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.pipeline import FeatureUnion, make_pipeline
from sklearn.svm import LinearSVC
word_vectorizer = TfidfVectorizer(ngram_range=(1, 2))
features = FeatureUnion([('words', word_vectorizer), ])
calibrated_svc = CalibratedClassifierCV(LinearSVC(), method='sigmoid', cv=3)
pipeline = make_pipeline(features, calibrated_svc)
pipeline.fit(train_x, train_y)
predicted = pipeline.predict_proba(test_x)
或使用逻辑回归:
from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.pipeline import FeatureUnion, make_pipeline
from sklearn.linear_model import LogisticRegression
word_vectorizer = TfidfVectorizer(ngram_range=(1, 2))
features = FeatureUnion([('words', word_vectorizer), ])
pipeline = make_pipeline(features, LogisticRegression())
pipeline.fit(train_x, train_y)
predicted = pipeline.predict_proba(test_x)
推荐阅读
- javascript - 使用 Jquery/Javascript 在特定页面上隐藏带有特定文本的特定 div
- javascript - 如何遍历数组以将它们组合成一个 JavaScript 对象?
- python - 在 Django 中不断收到“外键约束失败”
- ruby-on-rails - 为什么 ruby-jwt 在将 gem 从 1.3.0 更新到 2.2.2 后停止解码 JWT?
- java - 如何用 JMH 测量平均冷启动时间?
- c# - 使用 NPOI 包导出 excel 文件
- c++ - 我无法在 Visual Studio 2019 中访问我的 C++ 程序文件
- flutter - 列表/数组初始化的更好实践
- c++ - c ++多维地图:如何尊重第二个关键?
- laravel - 通过 laravel-socialite 后端的社交帐户登录用户