关于带有TensorFlow后端的Keras的自定义损失功能的问题

Question

我正在尝试将三类分类问题的损失函数定义如下：

def func_loss(y_true, y_pred):
    return -K.mean(K.prod(K.cast(K.argmax(y_pred, axis=1), K.floatx()) - 1.0, K.cast(K.argmax(y_true, axis=1), K.floatx()) - 1.0))

我的y看起来像这样： [[1,0,0], [0,1,0], [1,0,0], [0,0,1], ...]

直观地讲，我的三类标签是“ -1”，“ 0”和“ +1”类的一键编码。 我想最大化“ +/- 1”的正确标签，最小化“ +/- 1”的错误标签，并忽略所有“ 0”标签，无论它们是否正确。

当我使用此损失函数编译模型时，得到的是：

Traceback (most recent call last):
File "", line 1, in 
File "/usr/local/lib/python2.7/dist-packages/keras/models.py", line 547, in compile
**kwargs)
File "/usr/local/lib/python2.7/dist-packages/keras/engine/training.py", line 622, in compile
sample_weight, mask)
File "/usr/local/lib/python2.7/dist-packages/keras/engine/training.py", line 324, in weighted
score_array = fn(y_true, y_pred)
File "", line 2, in func_loss
File "/usr/local/lib/python2.7/dist-packages/keras/backend/tensorflow_backend.py", line 464, in prod
axis = _normalize_axis(axis, ndim(x))
File "/usr/local/lib/python2.7/dist-packages/keras/backend/tensorflow_backend.py", line 435, in _normalize_axis
if axis is not None and axis < 0:
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/ops.py", line 547, in nonzero
raise TypeError("Using a tf.Tensor as a Python bool is not allowed. "
TypeError: Using a tf.Tensor as a Python bool is not allowed. Use if t is not None: instead of if t: to test if a tensor is defined, and use TensorFlow ops such as tf.cond to execute subgraphs conditioned on the value of a tensor.

我一直在尝试使用此损失函数进行一些小的调整，但是在编译模型时，所有这些调整都会导致某些错误。 我想我可能对这件事的运作方式有一些基本的误解。 有人可以帮忙吗？

---

编辑：新的损失函数：

def func_loss(y_true, y_pred):
    return -K.mean((K.cast(K.argmax(y_pred, axis=1), K.floatx()) - 1.0 )* (K.cast(K.argmax(y_true, axis=1), K.floatx()) - 1.0))

Answer 1

我也在Tensorflow Github上回复了您。

您已经编辑了损失函数，但是实际上您现在遇到的错误略有不同。

之前，错误发生在K.mean

File "/usr/local/lib/python2.7/dist-packages/keras/backend/tensorflow_backend.py", line 490, in mean
axis = _normalize_axis(axis, ndim(x))

现在，您可以看到，它发生在K.prod

File "/usr/local/lib/python2.7/dist-packages/keras/backend/tensorflow_backend.py", line 464, in prod
axis = _normalize_axis(axis, ndim(x))

但是，原因仍然相同： K.mean和K.prod各自采用一个Tensor，因此，您传递的第二个Tensor被视为axis参数。