It gives me this error:
InvalidArgumentError: `predictions` contains negative values
Condition x >= 0 did not hold element-wise:
x (shape=(20000, 10) dtype=int64) = 
['3', '3', '-8', '...']

It has no sense because my labels are just 0 and 1.
This iwhat pred_labels has
[[  6.11138     2.9243512 -11.660926  … -11.982912  -12.400366
-12.061557 ]
[  6.1406865   2.6330147 -11.452074  … -11.73517   -12.167985
-11.924534 ]
[  6.0676413   2.5402145 -11.498982  … -11.899355  -12.084745
-11.687552 ]
…
[  6.1307297   2.6329107 -11.447449  … -11.732571  -12.161408
-11.918484 ]
[  6.056893    3.654493  -10.960058  … -11.586772  -11.884876
-11.678584 ]
[  6.1401978   2.6324804 -11.449804  … -11.733837  -12.1662035
-11.92291  ]]
It seems to be one image not predicted labels