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

时间:2016-05-17 17:43:40

标签: machine-learning scikit-learn tensorflow skflow

我尝试使用TensorFlow制作回归模型,同时使用sklearn实现,因此它与我所做的所有其他模型完美搭配。但是我似乎找不到用自定义评分函数(成本函数或目标函数)训练模型的方法。

使用skflow这根本不可能吗?

感谢加载!

1 个答案:

答案 0 :(得分:1)

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