Jeu de données PyTorch
Il est temps de rafraîchir vos connaissances sur les Datasets de PyTorch!
Avant de pouvoir entraîner un modèle, vous devez charger les données et les transmettre au modèle dans le bon format. Dans PyTorch, cela est géré par les Datasets et les DataLoaders. Commençons par construire un Dataset PyTorch pour nos données de potabilité de l'eau.
Dans cet exercice, vous allez définir une classe appelée WaterDataset pour charger les données à partir d'un fichier CSV. Pour ce faire, vous devez implémenter les trois méthodes qu'un Dataset PyTorch doit avoir :
.__init__()pour charger les données,.__len__()pour retourner la taille des données,.__getitem()__pour extraire les caractéristiques et l'étiquette d'un seul exemple.
Les importations suivantes dont vous avez besoin ont déjà été faites pour vous :
import pandas as pd
from torch.utils.data import Dataset
Cette activité fait partie du cours
Apprentissage profond intermédiaire avec PyTorch
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
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
____ = ____.____