在深度学习中寻找混淆矩阵

Question

我正在寻找从深度学习模式创建混乱矩阵的方法。但是面对下面的错误。

代码：

if __name__ == "__main__":
    print(y_test)
    X_new = tokenization_tweets(X_test, 3500)
    new_prediction = model_final.predict(X_new)
    print(new_prediction)

matrix = sklearn.metrics.confusion_matrix(y_test.argmax(axis=1), new_prediction.argmax(axis=1))

错误：

ValueError：axis必须小于维数（1）

任何帮助将不胜感激！