# 建立神经网络模型 model = keras.Sequential([ keras.layers.Flatten(input_shape=(28, 28)), # 将输入数据的形状进行修改成神经网络要求的数据形状 keras.layers.Dense(128, activation=tf.nn.relu), # 定义隐藏层,128个神经元的网络层 keras.layers.Dropout(0.2), keras.layers.Dense(10, activation=tf.nn.softmax) # 10个类别的分类问题,输出神经元个数必须跟总类别数量相同 ])
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:TensorFlow keras dropout层 - Python技术站