ÎncepețiÎncepe gratuit

Set de date PyTorch

Hai să îți reîmprospătezi cunoștințele despre seturile de date PyTorch!

Înainte de a putea antrena un model, trebuie să încarci datele și să le transmiți modelului în formatul potrivit. În PyTorch, de acest lucru se ocupă clasele Dataset și DataLoader. Să începem prin a construi un Dataset PyTorch pentru datele noastre despre potabilitatea apei.

În acest exercițiu, vei defini o clasă numită WaterDataset pentru a încărca datele dintr-un fișier CSV. Pentru aceasta, va trebui să implementezi cele trei metode pe care PyTorch le așteaptă de la un Dataset:

  • .__init__() pentru a încărca datele,
  • .__len__() pentru a returna dimensiunea setului de date,
  • .__getitem()__ pentru a extrage caracteristicile și eticheta unui singur eșantion.

Importurile necesare au fost deja făcute pentru tine:

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

Acest exercițiu face parte din cursul

Deep Learning intermediar cu PyTorch

Vezi cursul

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

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
        ____ = ____.____
Editează și rulează codul