python-2.7 - 如何在 sklearn 中转换数据点以进行预测

Question

我在 sklearn 中构建了一个模型，并将其保存为 pickle 形式。然而，为了获得最好的结果，我必须删除功能、创建另一个功能、使用 one-hot 编码等。

现在，我得到了一个初始形式的新数据点，我需要做出预测。有没有简单的方法来转换数据点，以便我构建的模型可以进行预测？

Answer 1

您需要对测试集应用与训练集相同的转换。实现此目的的方法之一是编写一个函数来处理数据并使用它。所以你会得到这样的东西:

def process_data(x):
    # process data, remove features and so on

     return processed_data

X_train_processed = process(X_train)
X_test_processed = process(X_test)