Začněte nyníZačněte zdarma

PyTorch Dataset

Čas si osvěžit znalosti PyTorch Datasetů!

Než začne trénování modelu, je potřeba data načíst a předat modelu ve správném formátu. V PyTorchi se o to starají Datasety a DataLoadery. Začneme vytvořením PyTorch Datasetu pro data o kvalitě pitné vody.

V tomto cvičení definuješ třídu WaterDataset, která načte data z CSV souboru. K tomu budeš muset implementovat tři metody, které PyTorch od Datasetu očekává:

  • .__init__() pro načtení dat,
  • .__len__() pro vrácení velikosti dat,
  • .__getitem()__ pro extrakci příznaků a štítku pro jeden vzorek.

Následující importy už jsou připravené za tebe:

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

Toto cvičení je součástí kurzu

Intermediate Deep Learning with PyTorch

Zobrazit kurz

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

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
        ____ = ____.____
Upravit a spustit kód