Применение пользовательской функции затрат в обучении модели SKFlow компании TensorFlow

Я пытаюсь создать регрессионную модель с помощью TensorFlow, используя реализацию sklearn, чтобы она хорошо сочеталась со всеми другими моими моделями. Однако я не могу найти способ обучения модели с помощью пользовательской функции оценки (функция стоимости или целевая функция).

Это просто невозможно с помощью skflow?

Спасибо огромное!

1 ответ

Решение

Многие из примеров использования learn.models.logistic_regression, которая в основном является встроенной моделью высокого уровня, которая возвращает прогнозы и потери. Например, models.logistic_regression использования ops.losses_ops.softmax_classifierЭто означает, что вы можете посмотреть, как ops.losses_ops.softmax_classifier реализована и реализована ваша собственная функция потерь с использованием, возможно, низкоуровневых API TensorFlow.

Другие вопросы по тегам