bojone / Capsule

A Capsule Implement with Pure Keras
350 stars 108 forks source link

当我用python3执行的时候遇到一个问题 #11

Open dashujuzha opened 5 years ago

dashujuzha commented 5 years ago

执行这一句的时候 greater = np.sort(Y_pred, axis=1)[:,-2] > 0.5 #判断预测结果是否大于0.5

遇到一个问题 numpy.core._internal.AxisError: axis 1 is out of bounds for array of dimension 1

不知道怎么解决了。谢谢。

bojone commented 5 years ago

我也不知道

matak07 commented 5 years ago

@dashujuzha Did you find the reason or solution of the error because even I am getting the same error by running the code with both python 2.7 and 3.5??