始める無料で始める

PyTorch Dataset

PyTorch の Dataset について復習しましょう！

モデルの学習を始める前に、データを読み込み、正しい形式でモデルに渡す必要があります。PyTorch では、これを Datasets と DataLoaders が担います。まずは、水の飲用適性データ用の PyTorch Dataset を作成していきます。

この演習では、CSV ファイルからデータを読み込むクラス WaterDataset を定義します。そのために、PyTorch の Dataset に求められる3つのメソッドを実装します。

  • データを読み込む .__init__()、
  • データサイズを返す .__len__()、
  • 単一サンプルの特徴量とラベルを取り出す .__getitem()__。

以下の必要なインポートはすでに用意されています。

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

この演習はコースの一部です

PyTorchによる中級ディープラーニング

コースを見る

実践的なインタラクティブ演習

このサンプルコードを完成させて、この演習に挑戦してみましょう。

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
        ____ = ____.____
コードを編集して実行