1. Learn
    2. /
  2. Курси
    3. /
  3. Середній рівень Deep Learning з PyTorch
Connected

вправа

Набір даних PyTorch

Час освіжити знання про PyTorch Datasets!

Перш ніж розпочати навчання моделі, потрібно завантажити дані та передати їх моделі у правильному форматі. У PyTorch за це відповідають Datasets і DataLoaders. Почнімо зі створення PyTorch Dataset для наших даних про придатність води до вживання.

У цій вправі ви визначите клас WaterDataset для завантаження даних із CSV-файла. Для цього потрібно реалізувати три методи, яких PyTorch очікує від Dataset:

  • .__init__() — завантаження даних,
  • .__len__() — повернення розміру даних,
  • .__getitem()__ — вибір ознак та мітки для одного зразка.

Необхідні імпорти вже виконано за вас:

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

Інструкції 1/3

undefined XP
    1
    2
    3
  • У методі .__init__() завантажте дані з csv_path у DataFrame pandas і присвойте його змінній df.
  • Перетворіть df на масив NumPy і запишіть результат у self.data.