ValueError: model_fn не принимает метки, но input_fn возвращает метки

Question

Я строю классификацию моделей, используя тензор потока-концентратора, тензор потока-оценки и тензор потока-данных.

Моя функция поезда возвращает набор данных и 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, ...)

Очень похоже на код Дэмиена.

Среда кода: Python 2, облачные данные Google, tf.version это 1.12. Ошибка, которая выдается model_fn не ожидает параметр меток (который, вероятно, генерируется tf-data Набор данных). Какой должна быть подпись model_fn При условии input_fn возвращает набор данных?

Посоветуйте пожалуйста с любой идеей.

Большое спасибо,

eilalan