我正在尝试将此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,)
我被困在如何为此创建自定义估算器。