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



我正在尝试在使用sklearn实现的同时使用TensorFlow制作回归模型,这样它就可以很好地与我制作的所有其他模型配合使用。然而,我似乎找不到一种方法来训练具有自定义评分函数(成本函数或目标函数)的模型。

这对skflow来说根本不可能吗?

非常感谢!

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

相关内容

  • 没有找到相关文章

最新更新