Anwenden von benutzerdefinierten Kostenfunktion in TensorFlow's SKFlow-Modelltraining

Question

Ich versuche, ein Regressionsmodell mit TensorFlow zu erstellen, während ich die sklearn-Implementierung verwende, damit es gut mit all meinen anderen Modellen zusammenspielt. Ich finde jedoch keine Möglichkeit, das Modell mit einer benutzerdefinierten Bewertungsfunktion (Kostenfunktion oder Zielfunktion) zu trainieren.

Ist das mit skflow einfach nicht möglich?

Vielen Dank!

Answer 1

Viele der Beispiele verwenden learn.models.logistic_regression, was im Grunde ein integriertes High-Level-Modell ist, das Vorhersagen und Verluste zurückgibt. Zum Beispiel verwendet ops.losses_ops.softmax_classifier, was bedeutet, dass Sie prüfen können, wie ops.losses_ops.softmax_classifier implementiert ist, und implementieren Sie Ihre eigene Verlustfunktion, die vielleicht TensorFlow Low-Level-APIs verwendet.