带有 ID 列的训练模型答案

【问题标题】：Training model with ID column带有 ID 列的训练模型
【发布时间】：2018-09-27 23:59:14
【问题描述】：

我正在使用 scikit-learn 训练一个模型，其中我的数据集中有一个 ID 列。我在训练模型时删除了 ID 列。但是对于测试数据集，我需要在进行预测后将其映射回 ID 列。

最好的方法是什么？我们可以在 scikit-learn 中构建模型时设置一个非预测器列吗？此外，其他 ML 工具（如 TensorFlow、Spark ML）一般情况如何。他们支持这个功能吗？

我在 stackoverflow 上找到了这个 post，但正在寻找其他选项。

【问题讨论】：

【解决方案1】：

我假设您将数据 (X) 存储在 pd.DataFrame 中。如果是这种情况，只需将值提取到 numpy ndarray 中。相应的行将具有相同的顺序。一个 scikit-learn 程式化的例子：

output = pd.Series(data=some_model.predict(X.values), index=X.index)

【讨论】：