Keras中怎么添加Layer到模型

小億
93
2024-03-19 15:23:35

在Keras中，可以使用模型的add()方法來(lái)添加層到模型中。例如，可以通過(guò)以下方式添加一個(gè)全連接層到模型中：

from keras.models import Sequential
from keras.layers import Dense

model = Sequential()
model.add(Dense(units=64, activation='relu', input_shape=(100,)))

在上面的例子中，我們首先創(chuàng)建了一個(gè)Sequential模型，然后通過(guò)add()方法添加了一個(gè)包含64個(gè)神經(jīng)元和ReLU激活函數(shù)的全連接層。輸入形狀為(100,)，表示輸入數(shù)據(jù)的維度為100。

除了Dense層，Keras還提供了其他類(lèi)型的層，例如卷積層、池化層、循環(huán)層等，可以根據(jù)需要選擇適合的層類(lèi)型添加到模型中。添加層到模型后，可以通過(guò)調(diào)用compile()方法來(lái)編譯模型，然后使用fit()方法來(lái)訓(xùn)練模型。

0 踩