keras吧 关注:224贴子:411
  • 0回复贴,共1

请教各位好心人,训练集和模型的input_shape到底应该怎么设定呀

只看楼主收藏回复

第一层:Conv2D(32, (3, 3), input_shape=(28, 28, 1))
x_train.shape = (60000, 28, 28), y_train.shape = (60000, 1)
那么 model.fit(x_train, y_train)之前,需要reshape一下x_train和y_train吗


IP属地:天津1楼2021-09-18 15:18回复