ПочатиПочніть безкоштовно

Використання TensorDataset

Структурування даних у датасет — один із перших кроків під час навчання нейронної мережі в PyTorch. TensorDataset спрощує це, перетворюючи масиви NumPy у формат, який може використовувати PyTorch.

У цій вправі ви створите TensorDataset, використовуючи попередньо завантажений набір даних animals, і перевірите його структуру.

Ця вправа є частиною курсу

Вступ до Deep Learning з 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)
Редагувати та запускати код