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
Anleitung zur Übung
- Wandle
X
undy
in Tensoren um und erstelle einTensorDataset
. - 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)