PyTorchでCSVデータセットを読み込むにはどうすればいいですか？

PyTorchでCSVデータセットを読み込むには、Pythonのpandasライブラリを使用してCSVファイルをロードし、PyTorchテンソルに変換します。以下に簡単な例を示します。

import pandas as pd
import torch
# 读取CSV文件
data = pd.read_csv('dataset.csv')
# 提取特征和标签列
features = data.iloc[:, :-1].values
labels = data.iloc[:, -1].values
# 将特征和标签转换为PyTorch张量
features_tensor = torch.tensor(features, dtype=torch.float32)
labels_tensor = torch.tensor(labels, dtype=torch.float32)
# 打印张量的形状
print("Features shape:", features_tensor.shape)
print("Labels shape:", labels_tensor.shape)

この例のなかでは、はじめにpandasのread_csv関数を用いてCSVファイルをロードしています。それから、.ilocを用いて特徴とラベルのカラムを抽出してnumpy配列に変換しています。最後に、torch.tensorを用いて特徴とラベルをPyTorchテンソルに変換しています。

注意、これは単なる基本的な例であり、実際には貴方のデータ・セットに特有な付加的なデータ前処理が必要になる場合があります。