PyTorch DataLoader

Skvělá práce při definování třídy Dataset! WaterDataset, který jsi právě vytvořil/a, je teď připravený k použití.

Dalším krokem při přípravě trénovacích dat je nastavení DataLoaderu. PyTorch DataLoader lze vytvořit z Datasetu – načítá data, rozděluje je do dávek a podle potřeby na nich provádí transformace. Výsledkem je vzorek dat připravený k trénování.

V tomto cvičení sestavíš DataLoader založený na WaterDatasetu. Třída DataLoader, kterou budeš potřebovat, je už naimportovaná z torch.utils.data. Pojďme na to!

Vytvoř instanci WaterDataset ze souboru water_train.csv a přiřaď ji do proměnné dataset_train.
Vytvoř dataloader_train na základě dataset_train s velikostí dávky dva a s náhodným promícháním vzorků.
Získej dávku příznaků a štítků z DataLoaderu a vypiš ji.

.css-6su6fj{-webkit-flex-shrink:0;-ms-flex-negative:0;flex-shrink:0;}cvičení

Pokyny

cvičení