自动化的神经网络求解

偏向自动化的神经网络有大佬能指教一下吗？私我一下

import autonn

# 加载MNIST数据集
(x_train, y_train), (x_test, y_test) = autonn.load_mnist()

# 创建神经网络模型
model = autonn.Model()
model.add(autonn.layers.Conv2D(32, kernel_size=(3, 3), activation="relu", input_shape=(28, 28, 1)))
model.add(autonn.layers.MaxPooling2D(pool_size=(2, 2)))
model.add(autonn.layers.Flatten())
model.add(autonn.layers.Dense(10, activation="softmax"))

# 编译模型
model.compile(optimizer="adam", loss="categorical_crossentropy", metrics=["accuracy"])

# 训练模型
model.fit(x_train, y_train, batch_size=128, epochs=10, validation_data=(x_test, y_test))