machine-learning - 在 TensorFlow 的 SKFlow 模型训练中应用自定义成本函数

Question

我正在尝试在使用 sklearn 实现的同时使用 TensorFlow 制作回归模型，因此它可以与我制作的所有其他模型很好地配合使用。但是，我似乎找不到使用自定义评分函数（成本函数或目标函数）训练模型的方法。

这对skflow来说是不可能的吗？

感谢负载！

Answer 1

许多示例使用learn.models.logistic_regression，它基本上是一个内置的高级模型，可以返回预测和损失。例如，models.logistic_regressionuses ops.losses_ops.softmax_classifier，这意味着您可以研究如何ops.losses_ops.softmax_classifier实现并使用 TensorFlow 低级 API 实现您自己的损失函数。