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

Đánh giá hiệu suất của mô hình

Nhóm PyBooks đang đạt được những bước tiến với công cụ gợi ý sách. Nhóm mô hình đã cung cấp cho bạn hai mô hình khác nhau, sẵn sàng cho công cụ gợi ý sách tại PyBooks. Một mô hình dựa trên LSTM (lstm_model) và mô hình còn lại dùng GRU (gru_model). Nhiệm vụ của bạn là đánh giá và so sánh các mô hình này.

Nhãn kiểm thử y_test và các dự đoán của mô hình gồm y_pred_lstm cho lstm_model và y_pred_gru cho gru_model.

Hướng dẫn

100 XP
  • Định nghĩa accuracy, precision, recall và F1 cho phân loại đa lớp bằng cách chỉ định num_classes và task.
  • Tính và in accuracy, precision, recall và F1 cho lstm_model.
  • Tương tự, tính các chỉ số đánh giá cho gru_model.