Использование TensorDataset
Структурирование данных в набор данных — один из первых шагов при обучении нейронной сети в PyTorch. TensorDataset упрощает этот процесс, преобразуя массивы NumPy в формат, с которым работает PyTorch.
В этом упражнении вы создадите TensorDataset на основе заранее загруженного набора данных animals и изучите его структуру.
Это упражнение является частью курса
Введение в глубокое обучение с PyTorch
Инструкции к упражнению
- Преобразуйте
Xиyв тензоры и создайтеTensorDataset. - Получите и выведите первый пример.
Интерактивное практическое упражнение
Попробуйте выполнить это упражнение, дополнив этот пример кода.
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)