通过CloudML获取TFrecord的批量预测

通过CloudML获取TFrecord的批量预测

导出模型时，需要确保它是“可批量的”，即输入占位符的外部尺寸具有

shape=[None]

，例如，

input = tf.Placeholder(dtype=tf.string, shape=[None])...

这可能需要重新处理图形。例如，我看到输出的形状被硬编码为[1,1]。最外面的尺寸应该是

None

，这在您固定占位符时可能会自动发生，或者可能需要进行其他更改。

给定输出的名称是

probabilities

，我还希望最里面的维度> 1，即要预测的类数，因此类似

[None, NUM_CLASSES]

。