Да уж. Вам нужно будет сохранить модель с несколькими головками. Взгляните на образец переписи тензора ядра двигателя (без оценки)

я изо всех сил пытался понять, каковы основные задачи serve_input_fn (), когда обученная модель экспортируется в Tensorflow для служебных целей. В Интернете есть несколько примеров, которые объясняют это, но у меня проблемы с определением для себя.

Проблема, которую я пытаюсь решить, - это проблема регрессии, где у меня 29 входов и один выход. Есть ли шаблон для создания соответствующей функции входного обслуживания для этого? Что если я использую задачу классификации одного класса? Должна ли быть изменена моя входная функция подачи или я могу использовать ту же функцию?

И наконец, мне всегда нужно обслуживать функции ввода или это только когда я использую tf.estimator для экспорта моей модели?

Ответы на вопрос(1)

Ваш ответ на вопрос