1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Deep Learning cho Văn bản với PyTorch
Connected

Bài tập

Xây dựng mô hình GRU cho văn bản

Tại PyBooks, đội ngũ rất ấn tượng với hiệu năng của hai mô hình bạn đã huấn luyện trước đó. Tuy nhiên, để đạt chất lượng cao nhất, họ muốn đảm bảo lựa chọn được mô hình tốt nhất cho nhiệm vụ hiện tại. Vì vậy, họ đề nghị bạn mở rộng dự án bằng cách thử nghiệm khả năng của các mô hình GRU, vốn nổi tiếng về hiệu quả và hiệu suất trong các bài toán phân loại văn bản. Nhiệm vụ mới của bạn là áp dụng mô hình GRU để phân loại các bài viết từ bộ dữ liệu Newsgroup vào các nhóm sau:

rec.autos, sci.med, và comp.graphics.

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

Hướng dẫn

100 XP
  • Hoàn thiện lớp GRU với các tham số cần thiết.
  • Khởi tạo mô hình với cùng các tham số đó.
  • Huấn luyện mô hình: truyền tham số vào hàm criterion và lan truyền ngược hàm mất mát.