Kom igångKom igång gratis

Använda TensorDataset

Att strukturera din data i en datamängd är ett av de första stegen när du tränar ett neuralt nätverk i PyTorch. TensorDataset förenklar det här genom att konvertera NumPy-arrayer till ett format som PyTorch kan använda.

I den här övningen skapar du en TensorDataset med den förinstallerade datamängden animals och undersöker dess struktur.

Den här övningen är en del av kursen

Introduktion till djupinlärning med PyTorch

Visa kurs

Övningsinstruktioner

  • Konvertera X och y till tensorer och skapa en TensorDataset.
  • Hämta och skriv ut det första samplet.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

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)
Redigera och kör kod