1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Nhập môn Deep Learning với PyTorch
Connected

Bài tập

Tạo nhãn one-hot

One-hot encoding chuyển một nhãn nguyên đơn thành một vector có N phần tử, trong đó N là số lượng lớp. Vector này chứa các số không và một số một ở vị trí tương ứng.

Trong bài tập này, bạn sẽ tự tạo một vector one-hot cho y, sau đó dùng PyTorch để đơn giản hóa quy trình. Tập dữ liệu của bạn có ba lớp (0, 1, 2).

numpy (np), torch.nn.functional (F), và torch đã được nhập sẵn cho bạn.

Hướng dẫn

100 XP
  • Tự mã hóa one-hot nhãn đúng y bằng mảng NumPy đã cho và lưu thành one_hot_numpy.
  • Dùng PyTorch để mã hóa one-hot y và lưu thành one_hot_pytorch.