Набор данных 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
____ = ____.____