Применение пользовательской функции затрат в обучении модели 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.