获取numpy.AxisError:轴2超出了维2数组的界限？python
获取numpy.AxisError:轴2超出了维2数组的界限？python
提问于 2022-10-19 11:08:24
我正在尝试建立一个机器学习模型：
def build_model(model,mlb_estimator,xtrain,ytrain,xtest,ytest):
clf = mlb_estimator(model)
clf.fit(xtrain,ytrain)
clf_predictions = clf.predict(xtest)
y_test_arg=np.argmax(ytest,axis=2,)
Y_pred = np.argmax(model.predict(Xtest),axis=1)
print('Confusion Matrix')
print(confusion_matrix(y_test_arg, Y_pred))
return result但是，在运行模型时，我会得到以下错误：
numpy.AxisError: axis 2 is out of bounds for array of dimension 2在第37行，即上面的第4行：
y_test_arg=np.argmax(ytest,axis=2,)我做错什么了？
回答 1
Stack Overflow用户
发布于 2022-10-19 12:06:53
您正在尝试访问二维数组的第三维空间(索引2在索引以0开始时)。将axis=2改为axis=1。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/74124306
复制相关文章
相似问题
腾讯云开发者
