在Estimator Tensorflow中使用“权重”

时间:2018-01-08 19:55:44

标签: python-3.x tensorflow

我已实施了一个自定义Estimator进行分类。我的input_fn函数返回features,labelsfeatures包含weights列,我希望在我的估算工具中将其用于我的损失函数中。

def my_model(features, labels, mode, params):
"""DNN with hidden layers."""

net = tf.feature_column.input_layer(features, params['feature_columns'])
weight = HOW TO GET `weight`FROM `features`

TF版本:1.4

0 个答案:

没有答案