Huấn luyện mô hình mạng nơ-ron trên dữ liệu quần áo
Trong bài tập này, bạn sẽ huấn luyện mạng nơ-ron fully connected mà bạn đã xây dựng ở bài trước trên dữ liệu ảnh. Dữ liệu huấn luyện được cung cấp qua hai biến: train_data chứa dữ liệu pixel cho 50 ảnh thuộc ba lớp quần áo và train_labels, chứa các vector one-hot cho nhãn của từng ảnh trong số 50 ảnh này. Hãy chuyển đổi dữ liệu thành định dạng đầu vào mà mạng mong đợi rồi huấn luyện mô hình trên dữ liệu và nhãn huấn luyện.
model bạn đã biên dịch ở bài trước cùng với train_data và train_labels đã có sẵn trong workspace 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
- Chuẩn bị dữ liệu để huấn luyện bằng cách reshape lại.
- Huấn luyện mô hình bằng cách truyền dữ liệu huấn luyện đầu vào và nhãn huấn luyện vào phương thức
.fit()của mô hình.
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.
# Reshape the data to two-dimensional array
train_data = train_data.reshape(____, ____)
# Fit the model
model.fit(____, ____, validation_split=0.2, epochs=3)