将现有的Tensorflow迁移到Estimator

时间:2018-10-25 07:31:31

标签: python tensorflow

我正在尝试将此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,)

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

0 个答案:

没有答案