在学习时忽略一列,但保留它以供SKLearn稍后参考

时间:2018-02-07 00:10:05

标签: scikit-learn attributes ignore

sklearn中是否有办法在测试/训练集中保留一列但忽略它进行学习? 我们假设我有一个名为" ID"和其他一些人一起我想忽略" ID"在构建模型时使用其他列,但有" ID"在预测结果中。我需要能够确定哪个预测结果属于哪个" ID"。

1 个答案:

答案 0 :(得分:0)

一种解决方案是使用带有步骤的pipeline选择要训练的列。

可以在scikit-learn文档中找到一些示例: