ValueError: cannot reshape array of size 1 into shape (28,28)
import matplotlib.pyplot as plt plt.figure(figsize=(10,8)) for idx, n in enumerate(samples): plt.subplot(4, 4, idx+1) plt.imshow(X_train[n].reshape(28,28), cmap='Greys') plt.title(y_train[n]) plt.show() mnist 데이터를 사용하고, random하게 16개를 뽑아서 그림을 그려보는 도중 위 코드에서 발생한 에러이다. random으로 뽑아낸 samples 데이터를 보면 숫자가 한줄로 나열되어있어서, 그림을 그릴 수 없기에 reshape(28,28)로 잡아주었는데, train 데이터를 보면 맨 앞에 label column이 있는데 X_train 지정시..
2023. 1. 15.
