MulaiMulai sekarang secara gratis

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

Lihat Kursus

Petunjuk latihan

  • Konversikan X dan y menjadi tensor dan buat TensorDataset.
  • 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)
Edit dan Jalankan Kode