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

Kiểm thử mô hình CNN phân tích cảm xúc

Giờ mô hình đã được huấn luyện, PyBooks muốn kiểm tra hiệu năng trên một số đánh giá sách mới.

Bạn cần kiểm tra cảm xúc trong một đánh giá là tích cực hay tiêu cực.

Các gói sau đã được import 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 tham số vocab_size và embed_dim cũng đã được tải và lưu dưới tên model.

Hướng dẫn

100 XP
  • Lặp qua danh sách book_reviews, chuyển các từ trong mỗi bài đánh giá thành một tensor.
  • Lấy đầu ra của mô hình cho mỗi input_tensor.
  • Tìm chỉ mục của hạng mục cảm xúc có khả năng cao nhất từ outputs.data.
  • Trích xuất và chuyển đổi phần tử predicted_label thành chuỗi cảm xúc, trong đó 1 là nhãn "Positive".