![](/img/trans.png)
[英]AxisError: axis 1 is out of bounds for array of dimension 1 when calculating roc_auc_score
[英]AxisError: axis 1 is out of bounds for array of dimension 1 when calculating accuracy of classes
我尝试使用此代码预测 10 个类
#Predicting the Test set rules
y_pred = model.predict(traindata)
y_pred = np.argmax(y_pred, axis=1)
y_true = np.argmax(testdata, axis=1)
target_names = ["akLembut","akMundur","akTajam","caMenaik", "caMenurun", "coretanTengah", "garisAtas", "garisBawah", "garisBawahBanyak", "ttdCangkang"]
print("\n"+ classification_report(y_true, y_pred, target_names=target_names))
但后来我收到这样的错误消息
AxisError Traceback (most recent call last)
<ipython-input-13-a2b02b251547> in <module>()
2 y_pred = model.predict(traindata)
3 y_pred = np.argmax(y_pred, axis=1)
----> 4 y_true = np.argmax(testdata, axis=1)
5
6 target_names = ["akLembut","akMundur","akTajam","caMenaik", "caMenurun", "coretanTengah", "garisAtas", "garisBawah", "garisBawahBanyak", "ttdCangkang"]
<__array_function__ internals> in argmax(*args, **kwargs)
2 frames
/usr/local/lib/python3.6/dist-packages/numpy/core/fromnumeric.py in _wrapit(obj, method, *args, **kwds)
45 except AttributeError:
46 wrap = None
---> 47 result = getattr(asarray(obj), method)(*args, **kwds)
48 if wrap:
49 if not isinstance(result, mu.ndarray):
AxisError: axis 1 is out of bounds for array of dimension 1
我已经训练了数据,我需要知道每一个准确度。
我的猜测是您的test_data
数组只是一维的,因此更改为y_true = np.argmax(testdata, axis=0)
。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.