Keras 混同行列

Question

前提・実現したいこと

Kerasを用いて画像認識を行っています。
そこで混同行列を表示しようとしているのですが、
以下のエラーが出てしまいどのようにすればよいかわかりません。
どのようにすればよいか教えていただけると幸いです。

クラス数は145クラス
https://newtechnologylifestyle.net/keras_fashion-mnist/
上のサイトを参考に作っています。

発生している問題・エラーメッセージ

 Singleton array 0 cannot be considered a valid collection.

該当のソースコード

predict_classes = model.predict_classes(X_test)
true_classes = np.argmax(y_test)
print(confusion_matrix(true_classes, predict_classes))

Answer 1

2行目の
true_classes = np.argmax(y_test)
を
true_classes = np.argmax(y_test, axis=1)
としたら表示することができました。