Python:DeprecationWarning:elementwise == 比较失败；这将在未来引发错误

Question

我在将模型的预测与训练集的标签进行比较时遇到了问题。我使用的数组具有以下形状:

Training set (200000, 28, 28) (200000,)
Validation set (10000, 28, 28) (10000,)
Test set (10000, 28, 28) (10000,)

但是，当使用函数检查准确性时:

def accuracy(predictions, labels):
    return (100.0 * np.sum(np.argmax(predictions, 1) == np.argmax(labels, 1))
          / predictions.shape[0])

它给了我:

C:\Users\Arslan\Anaconda3\lib\site-packages\ipykernel_launcher.py:5: DeprecationWarning: elementwise == comparison failed; this will raise an error in the future."""

它给出的所有数据集的准确率为 0%。

我认为我们不能使用“==”来比较数组。我怎样才能以正确的方式比较数组？

Answer 1

我假设错误发生在这个表达式中:

np.sum(np.argmax(predictions, 1) == np.argmax(labels, 1))

你能告诉我们一些关于 2 个数组、predictions、labels 的事情吗？通常的东西——dtype、shape、一些样本值。也许采取额外的步骤并为每个显示 np.argmax(...)。

在 numpy 中，您可以比较相同大小的数组，但比较大小不匹配的数组变得更加挑剔:

In [522]: np.arange(10)==np.arange(5,15)
Out[522]: array([False, False, False, False, False, False, False, False, False, False], dtype=bool)
In [523]: np.arange(10)==np.arange(5,14)
/usr/local/bin/ipython3:1: DeprecationWarning: elementwise == comparison failed; this will raise an error in the future.
  #!/usr/bin/python3
Out[523]: False