ÎncepețiÎncepe gratuit

Folosirea TensorDataset

Organizarea datelor într-un set de date este unul dintre primii pași în antrenarea unei rețele neuronale cu PyTorch. TensorDataset simplifică acest proces, convertind array-urile NumPy într-un format compatibil cu PyTorch.

În acest exercițiu, vei crea un TensorDataset folosind setul de date animals preîncărcat și vei inspecta structura acestuia.

Acest exercițiu face parte din cursul

Introducere în Deep Learning cu PyTorch

Vezi cursul

Instrucțiuni pentru exercițiu

  • Convertește X și y în tensori și creează un TensorDataset.
  • Accesează și afișează primul eșantion.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

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)
Editează și rulează codul