值错误:model fn不接受标签,但input fn返回标签

我正在使用tensorflow hub、tensorflow估计器和tensorflow数据建立模型分类。

我的train函数返回数据集 model_fn 定义如下:

def train_input_fn():
      return dataset_input_fn(DATASET_TRAIN_PATH)

def model_fn(features, labels, mode, params): 
  logging.info("model_fn")
  # module is imported from tf-hub
  return head.create_estimator_spec (features, mode, ...)

非常类似于 code by Damien .

代码环境是:python 2、google cloud datalab, tf.version 是1.12。 正在触发的错误是 模特儿 不需要labels参数(该参数可能由 tf-data 数据集)。签名应该是什么 模特儿 鉴于 input_fn 返回数据集?

请提出任何建议。

多谢,

埃拉兰