我在尝试使用TensorFlow构建一个回归模型,同时使用sklearn的实现,以便它能与我之前创建的其他模型兼容。然而,我似乎找不到一种方法来使用自定义评分函数(成本函数或目标函数)来训练模型。
这在skflow中根本不可能实现吗?
非常感谢!
回答:
许多示例使用了learn.models.logistic_regression
,这是一个内置的高级模型,用于返回预测和损失。例如,models.logistic_regression
使用了ops.losses_ops.softmax_classifier
,这意味着你可以研究ops.losses_ops.softmax_classifier
的实现方式,并使用TensorFlow的低级API来实现你自己的损失函数。