feature_columns的问题-feature_columns的项必须为_FeatureColumn

Question

大家好，我在Python 3.6中使用Tensorflow 1.12

尝试训练模型时遇到以下错误

ValueError：feature_columns的项必须为_FeatureColumn。给定的（type）：_NumericColumn（key ='a'，shape =（1，），default_value = None，dtype = tf.float32，normalizer_fn = None）。

我的input_fn定义如下

def mapper(x):
    return {"a":x.a, "b":x.b}, x.label

def input_fn():
    dataset = ... data
    dataset = dataset.batch(batch_size=32)
    features, labels = dataset.map(lambda x: mapper(x)).make_one_shot_iterator().get_next()
    return features, labels

我的功能如下

{'a': <tf.Tensor 'IteratorGetNext:0' shape=() dtype=int32>, 'b': <tf.Tensor 'IteratorGetNext:1' shape=() dtype=int32>}

我的feature_columns定义如下

my_feature_columns = []
for key in ['a', 'b']:
    my_feature_columns.append(tf.feature_column.numeric_column(key=key))

我正在寻找任何帮助！