1. Learn
    2. /
  2. Courses
    3. /
  3. Deep Learning cho Văn bản với PyTorch
Connected

Exercise

Huấn luyện một mô hình CNN cho văn bản

Bạn đã làm tốt khi định nghĩa lớp TextClassificationCNN. PyBooks giờ cần huấn luyện mô hình để tối ưu hóa việc phân tích cảm xúc của các bài đánh giá sách.

Các gói sau đã được nhập sẵn cho bạn: torch, torch.nn là nn, torch.nn.functional là F, torch.optim là optim.

Một thể hiện của TextClassificationCNN() với các đối số vocab_size và embed_dim cũng đã được nạp và lưu dưới tên model.

Instructions

100 XP
  • Định nghĩa một hàm mất mát dùng cho phân loại nhị phân và lưu là criterion.
  • Đưa gradient về 0 ở đầu vòng lặp huấn luyện.
  • Cập nhật các tham số ở cuối vòng lặp.