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

Connected

Bài tập

Embedding trong PyTorch

PyBooks đã thành công với hệ thống gợi ý sách, nhưng hệ thống này chưa tính đến một số ngữ nghĩa trong văn bản. Lớp embedding tích hợp sẵn của PyTorch có thể học và biểu diễn mối quan hệ giữa các từ trực tiếp từ dữ liệu. Nhóm của bạn muốn khám phá khả năng này để cải thiện hệ thống gợi ý sách. Bạn có thể giúp triển khai không?

torch và torch.nn dưới bí danh nn đã được import sẵn cho bạn.

Hướng dẫn

100 XP
  • Ánh xạ một chỉ số duy nhất cho mỗi từ trong words, lưu vào word_to_idx.
  • Chuyển word_to_idx thành một tensor của PyTorch và lưu vào inputs.
  • Khởi tạo một lớp embedding bằng mô-đun torch với mười chiều.
  • Truyền tensor inputs vào lớp embedding và xem kết quả đầu ra.