開始使用免費開始

訓練 CNN 來分類服裝種類

在訓練神經網路之前,必須先以正確的成本函式與適當的最佳化器進行編譯。編譯時,你也可以定義度量指標,讓網路在每個 epoch 計算並回報。模型擬合需要一組訓練資料,並將對應的訓練標籤一併提供給網路。

你在前一個練習中建立的 Conv2D model 已經在你的工作環境中可用。

本練習屬於課程

使用 Keras 進行影像建模

檢視課程

練習說明

  • 使用 'adam' 最佳化器與 'categorical_crossentropy' 成本函式來編譯網路。在 metrics 清單中定義網路需回報 'accuracy'
  • train_datatrain_labels 來擬合網路。訓練 3 個 epochs,批次大小為 10 張影像。訓練時,預留 20% 的資料作為驗證集,使用 validation_split 這個關鍵字引數。

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

# Compile the model 
model.compile(optimizer=____, 
              loss=____, 
              metrics=[____])

# Fit the model on a training set
model.fit(____, ____, 
          validation_split=____, 
          epochs=____, batch_size=____)
編輯並執行程式碼