6-5视频中,CNN的训练时出现问题

来源:6-5 卷积神经网络实战

幕布斯5473859

2019-08-19


http://img.mukewang.com/szimg/5d5a6922086e69cb09040637.jpg

写回答

1回答

正十七

2019-08-23

同学你好,看问题描述应该是你没做数据变换,我们需要把输入变成[batch_size, 28, 28, 1]的四维矩阵:

scaler = StandardScaler()
x_train_scaled = scaler.fit_transform(
    x_train.astype(np.float32).reshape(-1, 1)).reshape(-1, 28, 28, 1)
x_valid_scaled = scaler.transform(
    x_valid.astype(np.float32).reshape(-1, 1)).reshape(-1, 28, 28, 1)
x_test_scaled = scaler.transform(
    x_test.astype(np.float32).reshape(-1, 1)).reshape(-1, 28, 28, 1)


1
0

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

查看课程