将DNNRegressor模型导出到Cloud ML

Question

我试图使用该操作：

  

tf.contrib.learn.DNNRegressor.export（* args，** kwargs）

创建Google Cloud ML上传所需的export.meta文件。 我搜索了文档（和Google），但没有找到一个实际的例子。

我写了一个非常简单的回归量，它近似于两个数字之间的总和（小于一）：

# training data
def input_fn(size):
    A, B, R = [], [], []
    for x in range(size):
        a, b = random.random(), random.random()
        A.append(a)
        B.append(b)
        R.append(a + b)
    feature_cols = {'A': tf.constant(A, shape=[size, 1]),
                    'B': tf.constant(B, shape=[size, 1])}
    labels = tf.constant(R, shape=[size, 1])
    return feature_cols, labels

# regressor
FEATURES = ['A', 'B']
feature_cols = [tf.contrib.layers.real_valued_column(k) for k in FEATURES]
regressor = tf.contrib.learn.DNNRegressor(feature_columns=feature_cols, 
                                          hidden_units = [5, 3, 2],
                                          model_dir=MODEL_PATH)

# training
regressor.fit(input_fn=lambda: input_fn(1000), steps=5000)

#export
regressor.export(....

一切正常，但我无法构建导出功能所需的输入和输出映射器。 上传到Cloud ML时，我遇到了错误：

  

模型元图没有输出集合。

谢谢！