我尝试使用TensorFlow制作回归模型,同时使用sklearn实现,因此它与我所做的所有其他模型完美搭配。但是我似乎找不到用自定义评分函数(成本函数或目标函数)训练模型的方法。
使用skflow这根本不可能吗?
感谢加载!
答案 0 :(得分:1)
许多示例使用learn.models.logistic_regression
,它基本上是一个返回预测和损失的内置高级模型。例如,models.logistic_regression
使用ops.losses_ops.softmax_classifier
,这意味着您可以查看ops.losses_ops.softmax_classifier
的实现方式,并使用TensorFlow低级API实现您自己的损失函数。