我已实施了一个自定义Estimator进行分类。我的input_fn
函数返回features,labels
。 features
包含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