詢問Dropout問題

dropout model.add(dropout cnn

orga_an21 2022-06-28 14:41:05 ‧ 852 瀏覽

分享至

遇到NameError: name 'Dropout' is not defined
要怎麼處理才能讓Dropout被定義

model = models.Sequential()
model.add(layers.Conv2D(filters=64, kernel_size=3, activation='relu', 
                         input_shape=(IMG_SIZE, IMG_SIZE, 3)))
model.add(layers.Conv2D(filters=64, kernel_size=3, activation='relu'))
model.add(layers.MaxPool2D(2))
model.add(layers.MaxPooling2D(pool_size=(2, 2)))
model.add(Dropout(0.25))
model.add(layers.Flatten())
model.add(Dropout(0.5))              
model.add(layers.Dense(128, activation='relu'))            
model.add(layers.Dense(3, activation='softmax'))