【Python】模型优化与超参数选择

importtensorflowastfimportpandasaspdimportnumpyasnpimportmatplotlib.pyplotasplt# 模型优化与超参数选择(train_image, train_label), (test_image, test_label) =tf.keras.datasets.fashion_mnist.load_data()
train_label_onehot=tf.keras.utils.to_categorical(train_label)
test_label_onehot=tf.keras.utils.to_categorical(test_label)
model=tf.keras.Sequential()
model.add(tf.keras.layers.Flatten(input_shape=(28, 28)))
model.add(tf.keras.layers.Dense(128, activation='relu'))
model.add(tf.keras.layers.Dense(128, activation='relu'))
model.add(tf.keras.layers.Dense(128, activation='relu'))
model.add(tf.keras.layers.Dense(10, activation='softmax'))
model.compile(optimizer=tf.keras.optimizers.Adam(learning_rate=0.001), loss='categorical_crossentropy', metrics=['acc'])
model.fit(train_image, train_label_onehot, epochs=20)