Bắt đầu ngayBắt đầu miễn phí

Sử dụng TensorDataset

Cấu trúc dữ liệu của bạn thành một dataset là một trong những bước đầu tiên khi huấn luyện mạng nơ-ron PyTorch. TensorDataset giúp đơn giản hóa việc này bằng cách chuyển đổi mảng NumPy sang định dạng mà PyTorch có thể dùng.

Trong bài tập này, bạn sẽ tạo một TensorDataset từ bộ dữ liệu animals đã được nạp sẵn và kiểm tra cấu trúc của nó.

Bài tập này là một phần của khóa học

Nhập môn Deep Learning với PyTorch

Xem khóa học

Hướng dẫn bài tập

  • Chuyển X và y thành tensor và tạo một TensorDataset.
  • Truy cập và in ra mẫu đầu tiên.

Bài tập tương tác thực hành trực tiếp

Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.

import torch
from torch.utils.data import TensorDataset

X = animals.iloc[:, 1:-1].to_numpy()  
y = animals.iloc[:, -1].to_numpy()

# Create a dataset
dataset = ____(____, ____)

# Print the first sample
input_sample, label_sample = ____
print('Input sample:', input_sample)
print('Label sample:', label_sample)
Chỉnh sửa và Chạy Mã