Začněte nyníZačněte zdarma

Použití TensorDataset

Uspořádání dat do datasetu je jeden z prvních kroků při trénování neuronové sítě v PyTorchi. TensorDataset tento proces zjednodušuje tím, že převádí NumPy pole do formátu, se kterým PyTorch umí pracovat.

V tomto cvičení vytvoříš TensorDataset z předem načteného datasetu animals a prozkoumáš jeho strukturu.

Toto cvičení je součástí kurzu

Introduction to Deep Learning with PyTorch

Zobrazit kurz

Pokyny k cvičení

  • Převeď X a y na tensory a vytvoř TensorDataset.
  • Přistup k prvnímu vzorku a vypiš ho.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

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)
Upravit a spustit kód