Gradient Optimizer的自定义外部损失指标?

时间:2017-09-22 12:05:24

标签: optimization tensorflow loss

我有一个外部函数,它采用y和y_prediction(矩阵格式),并计算一个度量,它描述了预测的实际好坏。

不幸的是,度量标准不是简单的y - ypred或混淆矩阵,但仍然非常有用和重要。我如何使用这个为丢失计算的数字或作为optimizer.minimize的参数?

1 个答案:

答案 0 :(得分:0)

如果我理解正确,我认为有两种方法可以做到这一点:

您想要计算的损失可以写为tensorflow操作定义了渐变(例如,SVD没有在张量流库中定义渐变),然后优化是直接的。

或者您总是可以使用numpy运算符编写损失函数并使用tf.py_func()https://www.tensorflow.org/api_docs/python/tf/py_func,然后您必须手动显示渐变,如下所示:How to make a custom activation function with only Python in Tensorflow? 但你必须知道渐变的明确公式......