Menggunakan TensorDataset
Menyusun data Anda ke dalam sebuah himpunan data adalah salah satu langkah awal dalam melatih jaringan saraf PyTorch. TensorDataset menyederhanakan proses ini dengan mengonversi array NumPy ke format yang dapat digunakan PyTorch.
Pada latihan ini, Anda akan membuat TensorDataset menggunakan himpunan data animals yang sudah dimuat dan meninjau strukturnya.
Latihan ini adalah bagian dari kursus
Pengantar Deep Learning dengan PyTorch
Petunjuk latihan
- Konversikan
Xdanymenjadi tensor dan buatTensorDataset. - Akses dan cetak sampel pertama.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
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)