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

Đánh giá các mô hình phân loại RNN

Nhóm tại PyBooks muốn bạn đánh giá mô hình RNN mà bạn đã tạo và chạy với bộ dữ liệu Newsgroup. Nhớ rằng mục tiêu là phân loại các bài viết vào một trong ba nhóm:

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

Mô hình đã được huấn luyện và bạn đã in số epoch và loss cho mỗi mô hình.

Hãy dùng torchmetrics để đánh giá nhiều metric cho mô hình của bạn. Các metric sau đã được nạp sẵn: Accuracy, Precision, Recall, F1Score.

Một instance của rnn_model được huấn luyện ở bài trước cũng đã được nạp sẵn cho bạn.

Hướng dẫn

100 XP
  • Tạo một instance cho mỗi metric dành cho phân loại đa lớp với num_classes bằng số lượng danh mục.
  • Sinh dự đoán cho rnn_model bằng dữ liệu kiểm tra X_test_seq.
  • Tính các metric bằng các lớp dự đoán và nhãn đúng.