时间戳

首页 > 解决方案 > Sagemaker Keras 输出到 csv

keras - Sagemaker Keras 输出到 csv

问题描述

我的输入是一个csv文件：没有标题，第一列是id，其余的是特征

我正在使用 keras 预测具有 4 个变量的结果。我想将它与原始 ID 相关联。

我的最后一层keras看起来像：model.add(Dense(4, activation='linear'))

我没有 ID 的输出如下所示：（我相信它不是 csv 格式。）

{
    "predictions": [[1.23, 1.34, 2.42, 1.80], ...]
}

这似乎很奇怪，我什至无法加入输入。

有什么想法吗？提前致谢。

标签： kerasamazon-sagemaker

解决方案

您可以提供“inference.py”脚本作为模型类的入口点，并将预测输出保存到 csv 文件。

以下是有关如何使用预处理/后处理的说明：

https://github.com/aws/sagemaker-tensorflow-serving-container#prepost-processing

推荐阅读