继续上节课
我们要在 一个 简单的数据集 上, 训练一个 简单的ann模型;
代码
# 这是 第 5 节课的内容
model.compile(optimizer=Adam(learning_rate=0.0001), loss='sparse_categorical_crossentropy', metrics=['accuracy'])
model.fit(x=scaled_train_samples, y=train_lables, batch_size=10, epochs=30, shuffle=True, verbose=2)
# verbose=2 是看最多 output 的 级别;
实验结果