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