Один из ModeKeys. Указывает, является ли это обучение, оценка или предсказание.
predictions
Предсказания Tensor или словарь Tensor.
loss
Потеря обучения Tensor.
train_op
Операция для шага обучения.
eval_metric_ops
Словарь результатов метрик, индексированных по имени. Значения словаря — результаты вызова функции метрики, например, Tensor.
output_alternatives
Словарь {submodel_name: (problem_type, {tensor_name: Tensor})}, где submodel_name — идентификатор подмодели, который должен быть согласован на всём протяжении обработки данных (вероятно, взятый из имени каждой Head, для моделей, которые их используют), problem_type — ProblemType, tensor_name — символическое имя тензора вывода, возможно, но не обязательно, взятое из PredictionKey, и Tensor — соответствующий тензор вывода.
training_chief_hooks
Список объектов SessionRunHook, которые будут выполняться на главном рабочем узле во время обучения.
training_hooks
Список объектов SessionRunHook, которые будут выполняться на всех рабочих узлах во время обучения.
scaffold
Объект tf.compat.v1.train.Scaffold, который можно использовать для настройки инициализации, сохранения и других параметров, используемых в обучении.
Требуется для нескольких головок. Если у вас несколько элементов в словаре output_alternatives (аналогично нескольким головкам), EstimatorSpec требует головку по умолчанию, которая будет использоваться, если запрос Servo не указывает явно, на какую головку производить вывод. Передайте здесь ключ альтернативного вывода, который вы хотите назначить по умолчанию. Отдельная ExportOutpout для этой головки по умолчанию будет добавлена в словарь export_outputs со специальным ключом saved_model.DEFAULT_SERVING_SIGNATURE_DEF_KEY, если в output_alternatives ещё нет записи с этим специальным ключом.
Возвращаемое значение
Экземпляр EstimatorSpec , эквивалентный этому ModelFnOps