1. 学ぶ
    2. /
  2. コース
    3. /
  3. PyTorchによる中級ディープラーニング
Connected

演習

PyTorch DataLoader

Dataset クラスの定義、お見事です！ 先ほど作成した WaterDataset が使える状態になっています。

次のステップは、学習データの準備として DataLoader を設定することです。PyTorch の DataLoader は、Dataset から作成でき、データの読み込み、バッチ分割、必要に応じた変換の適用を行います。これにより、学習に使えるデータサンプルが順に取り出せます。

この演習では、WaterDataset に基づいた DataLoader を作成します。必要な DataLoader クラスはすでに torch.utils.data からインポート済みです。さっそく始めましょう！

指示

100 XP
  • water_train.csv から WaterDataset のインスタンスを作成し、dataset_train に代入します。
  • バッチサイズを 2、サンプルをシャッフルする設定で、dataset_train に基づく dataloader_train を作成します。
  • DataLoader から特徴量とラベルのバッチを取得し、出力してください。