在Keras中,模型的訓練通常通過調(diào)用模型的fit
方法來實現(xiàn)。fit
方法接受輸入數(shù)據(jù)和標簽,并根據(jù)給定的損失函數(shù)和優(yōu)化器來訓練模型。
以下是一個簡單的示例,演示如何在Keras中訓練一個簡單的神經(jīng)網(wǎng)絡模型:
import numpy as np
from keras.models import Sequential
from keras.layers import Dense
# 生成一些隨機的訓練數(shù)據(jù)
X_train = np.random.random((1000, 20))
y_train = np.random.randint(2, size=(1000, 1))
# 創(chuàng)建一個簡單的神經(jīng)網(wǎng)絡模型
model = Sequential()
model.add(Dense(64, input_dim=20, activation='relu'))
model.add(Dense(1, activation='sigmoid'))
# 編譯模型,定義損失函數(shù)和優(yōu)化器
model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
# 訓練模型
model.fit(X_train, y_train, epochs=10, batch_size=32)
在上面的示例中,我們首先生成了一些隨機的訓練數(shù)據(jù),然后創(chuàng)建了一個包含兩個全連接層的簡單神經(jīng)網(wǎng)絡模型。接著我們使用compile
方法來編譯模型,定義了損失函數(shù)為二分類交叉熵,優(yōu)化器為Adam,并指定了評估指標為準確率。最后,我們使用fit
方法來訓練模型,指定了訓練數(shù)據(jù)X_train
和標簽y_train
,并指定了訓練輪數(shù)和批處理大小。
希望這個示例能夠幫助你開始在Keras中訓練你的模型。