НачатьНачать бесплатно

Набор данных PyTorch

Пора освежить знания о наборах данных PyTorch!

Прежде чем начать обучение модели, необходимо загрузить данные и передать их модели в нужном формате. В PyTorch для этого используются Dataset и DataLoader. Начнём с создания набора данных PyTorch для данных о качестве воды.

В этом упражнении вы определите класс WaterDataset для загрузки данных из CSV-файла. Для этого нужно реализовать три метода, которые PyTorch ожидает от набора данных:

  • .__init__() — для загрузки данных,
  • .__len__() — для возврата размера данных,
  • .__getitem__() — для извлечения признаков и метки одного примера.

Следующие импорты уже выполнены за вас:

import pandas as pd
from torch.utils.data import Dataset

Это упражнение является частью курса

Глубокое обучение на PyTorch: средний уровень

Посмотреть курс

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

class WaterDataset(Dataset):
    def __init__(self, csv_path):
        super().__init__()
        # Load data to pandas DataFrame
        df = ____
        # Convert data to a NumPy array and assign to self.data
        ____ = ____.____
Редактировать и запускать код