神经网络六步搭建法
- import
- train , test
- model = tf.keras.models.Sequential( )
- model = compile( )
- model = fit ( )
- summary
笔记
第三步:神经网络结构。
model = tf.keras.models.Sequential([
tf.keras.layers.Flatten(),
tf.keras.layers.Dense(128, activation='relu'),
tf.keras.layers.Dense(10, activation='softmax')
])
第四步:神经网络配置。
model.compile(optimizer='adam',
loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=False),
metrics=['sparse_categorical_accuracy'])
第五步:训练参数。
model.fit(x_train, y_train, batch_size=1, epochs=4
, validation_data=(x_test, y_test), validation_freq=1)
根据六步法搭建神经网络,可以对神经网络有个大概认知,为后续深入学习打下基础。