Я строю классификацию моделей, используя тензор потока-концентратор, тензор потока-оценки и тензор потока-данных.
Моя функция поезда возвращает набор данных, и 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, ...)
Очень похоже на код от Damien .
Среда кода: Python 2, облачное хранилище данных Google, tf.version равно 1.12.
Возникающая ошибка: model_fn не ожидает метки (который, вероятно, генерируется набором данных tf-data). Какой должна быть подпись model_fn, если input_fn возвращает набор данных?
Посоветуйте, пожалуйста, любую идею.
Большое спасибо,
eilalan