我正在尝试建立一个机器学习模型：

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,)

我到底做错了什么？