Zacznij terazZacznij za darmo

Korzystanie z TensorDataset

Zorganizowanie danych w postaci zbioru danych to jeden z pierwszych kroków w trenowaniu sieci neuronowej w PyTorchu. TensorDataset upraszcza ten proces – konwertuje tablice NumPy na format obsługiwany przez PyTorch.

W tym ćwiczeniu utworzysz TensorDataset na podstawie wczytanego zbioru danych animals i przyjrzysz się jego strukturze.

To ćwiczenie jest częścią kursu

Wprowadzenie do uczenia głębokiego z PyTorch

Zobacz kurs

Instrukcje do ćwiczenia

  • Przekonwertuj X i y na tensory i utwórz TensorDataset.
  • Wyświetl pierwszą próbkę.

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

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)
Edytuj i uruchom kod