LoslegenKostenlos loslegen

TensorDataset verwenden

Die Strukturierung deiner Daten in einen Datensatz ist einer der ersten Schritte beim Training eines neuronalen Netzes von PyTorch. TensorDataset vereinfacht diesen Vorgang, indem es NumPy-Arrays in ein Format konvertiert, das PyTorch nutzen kann.

In dieser Übung erstellst du ein TensorDataset mit dem vorgeladenen animals-Datensatz und überprüfst seine Struktur.

Diese Übung ist Teil des Kurses

Einführung in Deep Learning mit PyTorch

Kurs anzeigen

Anleitung zur Übung

  • Wandle X und y in Tensoren um und erstelle ein TensorDataset.
  • Rufe die erste Probe auf und gib sie aus.

Interaktive Übung

Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.

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)
Code bearbeiten und ausführen