Erste SchritteKostenlos loslegen

TensorDatensatz verwenden

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

In dieser Übung erstellst du eine TensorDataset mit dem vorgeladenen animals Datensatz und überprüfst ihre 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 das erste Muster auf und drucke es aus.

Interaktive Übung zum Anfassen

Probieren Sie diese Übung aus, indem Sie diesen Beispielcode ausführen.

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)
Bearbeiten und Ausführen von Code