Huấn luyện một CNN sâu để phân loại ảnh quần áo
Huấn luyện một mô hình deep learning khá giống với huấn luyện mạng một tầng. Sau khi xây dựng mô hình (như bạn đã làm ở bài trước), bạn cần biên dịch mô hình với tập tham số phù hợp. Tiếp theo, huấn luyện mô hình bằng cách cung cấp dữ liệu huấn luyện và nhãn tương ứng. Khi huấn luyện xong, bạn có thể đánh giá mô hình trên dữ liệu kiểm thử.
model bạn đã xây dựng ở bài trước có sẵn trong không gian làm việc của bạn.
Bài tập này là một phần của khóa học
Mô hình hóa ảnh với Keras
Hướng dẫn bài tập
- Biên dịch mô hình với hàm mất mát categorical cross-entropy và bộ tối ưu Adam.
- Huấn luyện mạng với
train_datatrong 3 epoch, mỗi batch gồm 10 ảnh. - Trong quá trình huấn luyện, dùng ngẫu nhiên 20% dữ liệu huấn luyện làm dữ liệu xác thực.
- Đánh giá mô hình với
test_data, dùng kích thước batch là 10.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# Compile model
model.____(optimizer=____,
loss='categorical_crossentropy',
metrics=['accuracy'])
# Fit the model to training data
model.____(____, ____,
validation_split=0.2,
epochs=3, batch_size=10)
# Evaluate the model on test data
model.____(____, ____, batch_size=10)