我正在尝试在 sklearn 管道中添加校准步骤以获得校准的分类器,因此 have more trustworthy probabilities在输出中。
到目前为止,我笨拙地尝试使用 CalibratedClassifierCV 插入“校准”步骤沿着(重复性的愚蠢例子):
import sklearn.datasets
import pandas as pd
import numpy as np
from sklearn.pipeline import Pipeline
from sklearn.linear_model import SGDClassifier
from sklearn.feature_extraction.text import TfidfVectorizer
data = sklearn.datasets.fetch_20newsgroups(categories=['alt.atheism', 'sci.space'])
df = pd.DataFrame(data = np.c_[data['data'], data['target']])\
.rename({0:'text', 1:'class'}, axis = 'columns')
my_pipeline = Pipeline([
('vectorizer', TfidfVectorizer()),
('classifier', SGDClassifier(loss='modified_huber')),
('calibrator', CalibratedClassifierCV(cv=5, method='isotonic'))
])
my_pipeline.fit(df['text'].values, df['class'].values)
但这行不通(至少不是这样)。有没有人有关于如何正确执行此操作的提示?
最佳答案
SGDClassifier
对象应该进入 CalibratedClassifierCV
's base_estimator
argument .
您的代码可能看起来像这样:
my_pipeline = Pipeline([
('vectorizer', TfidfVectorizer()),
('classifier', CalibratedClassifierCV(base_estimator=SGDClassifier(loss='modified_huber'), cv=5, method='isotonic'))
])
CalibratedClassifierCV
是一个元估计器。
关于python - 在 scikit-learn 管道中插入 CalibratedClassifierCV 的正确方法是什么?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49832981/