1,导入所需要的模块。
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense, Flatten, Softmax
2,构建model。
有两种方式。第一种直接创建,将每一层的具体参数传递给Sequential().
model = Sequential([
Flatten(input_shape=(100,100)),
Dense(16,activation='relu'),
Dense(16,activation='relu'),
Dense(10,activation='softmax')
])
第二种,先创建个model对象,然后再需要增加层的时候,随时添加,比较灵活。
model = Sequential()
model.add(Flatten(input_shape=(50,50)))
model.add(Dense(16,activation='relu'))
model.add(Dense(16,activation='relu'))
model.add(Dense(10,activation='softmax'))
3,打印model的每一层的具体信息。
model.summary()