python - 错误 "TypeError: type numpy.ndarray doesn' t 定义 __round__ 方法”

Question

import numpy

......

# Prediction
predictions = model.predict(X_test)
# round predictions
rounded = [round(x) for x in predictions]
print(rounded)

"predictions" is a list of decimals between [0,1] with sigmoid output. 

为什么总是报这个错:

  File "/home/abigail/workspace/ml/src/network.py", line 41, in <listcomp>
    rounded = [round(x) for x in predictions]
TypeError: type numpy.ndarray doesn't define __round__ method

如果我不使用“圆”，它会正确打印小数。这个“圆”应该是 Python 内置函数。为什么它与 numpy 有任何关系？

编辑:

for x in predictions:
    print(x, end=' ')

输出是:

    [ 0.79361773] [ 0.10443521] [ 0.90862566] [ 0.10312044] [ 0.80714297] 
[ 0.23282401] [ 0.1730803] [ 0.55674052] [ 0.94095331] [ 0.11699325] 
[ 0.1609294] 

Answer 1

TypeError: type numpy.ndarray doesn't define round method

您尝试将轮次应用于 numpy.ndarray。显然，这不受支持。

试试这个，使用 numpy.round:

rounded = [numpy.round(x) for x in predictions]

x 是 numpy 数组。你也可以试试这个:

rounded = [round(y) for y in x for x in predictions]