将现有的Tensorflow迁移到Estimator

Question

我正在尝试将此TF pre 1.4 version转换为估算器，以在TF 1.10中使用。

这是主要功能：

distance = tf.reduce_sum(tf.abs(tf.add(xtr, tf.negative(xte))), reduction_indices=1)
# Prediction: Get min distance index (Nearest neighbor)
pred = tf.arg_min(distance, 0)

如何将其转换为自定义的估算器代码？

使用此列表：

• 定义功能列列表
• 创建估算器模型
• 创建数据输入函数
• 致电培训，评估和预测

我的功能列是： 每个样本是：

Features shape: (5000, 784)
Single Feature shape: (784,)

我被困在如何为此创建自定义估算器。