Tensorflow 带ID列的训练模型

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

最好的方法是什么？在scikit learn中构建模型时，我们可以设置非预测列？另外，其他的ML工具，比如TensorFlow，Spark ML，一般来说又如何呢。他们支持这个功能吗

---

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

我假设您将数据（X）存储在pd.DataFrame中。 如果是这种情况，只需将值提取到numpy ndarray中即可。相应的行将具有相同的顺序。scikit学习风格化示例：

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

---

只是在预测时不要发送ID列。输出将与输入匹配。