tensorflow - 带 ID 列的训练模型

Question

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

执行此操作的最佳方法是什么？我们可以在 scikit-learn 中构建模型时设置非预测列吗？另外，其他 ML 工具(例如 TensorFlow、Spark ML)又如何呢？他们支持这个功能吗？

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

Answer 1

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

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