如何使用 sklearn.pipeline 转换新数据

Posted 2023-03-12

【中文标题】如何使用 sklearn.pipeline 转换新数据

【英文标题】：How to transform new data using sklearn.pipeline

【发布时间】：2018-04-26 21:41:23

【问题描述】：

我用 TfIdfVectorizer 转换器和 OnevsRestClassifier 估计器创建了一个管道，并在训练数据上对其进行了如下训练

# Split data using train_test_split
print "Split data into train and test sets"
x_train, x_test, y_train, y_test = train_test_split(
    data_x, data_y, test_size=0.33)

# transform matrix of plots into lists to pass to a TfidfVectorizer
train_x = [x[0].strip() for x in x_train.tolist()]
test_x = [x[0].strip() for x in x_test.tolist()]

# Pipeline fit and transform
print "Learn the model using train data"
model = text_clf.fit(train_x, y_train)

# Predict the test data
print "Predict the recipients on test data"
predictions = model.predict(test_x)

现在，我想使用经过训练的模型来预测新的未标记数据的类别。 我试过了，报错了

# Read text from input
text = raw_input()
print "Input : ", text
new_data = text_clf.transform([text])
predict = model.predict(new_data) 

这是错误。我做错了什么？

AttributeError: 'OneVsRestClassifier' object has no attribute 'transform'

【参考方案1】：

如果text_clf 和model 是您建议的管道，则无需调用转换然后预测。只调用

predictions = model.predict([text]) 

管道将在内部自动将数据转换为可用格式（在中间转换器上使用transform()）。

当您显式调用 model.transform() 时，管道会假定管道内的所有估算器都有一个 transform()，但这里不是这种情况。