Mulai sekarangMulai 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 merupakan bagian dari kursus

Pengantar Deep Learning dengan PyTorch

Lihat Kursus

Instruksi latihan

  • Konversikan X dan y menjadi tensor dan buat TensorDataset.
  • Akses dan cetak sampel pertama.

Latihan interaktif langsung praktik

Cobalah latihan ini dengan melengkapi kode contoh ini.

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