中文字幕av专区_日韩电影在线播放_精品国产精品久久一区免费式_av在线免费观看网站

溫馨提示×

pytorch讀取csv數據集的方法有哪些

小億
182
2024-03-30 13:31:04
欄目: 深度學習

在PyTorch中讀取CSV數據集通常有以下幾種方法:

  1. 使用Pandas庫讀取CSV文件,并將其轉換為PyTorch張量:
import pandas as pd
import torch

# 讀取CSV文件
data = pd.read_csv('data.csv')

# 將數據轉換為PyTorch張量
tensor_data = torch.tensor(data.values)
  1. 使用PyTorch的Dataset和DataLoader類來讀取CSV文件:
import torch
from torch.utils.data import Dataset, DataLoader

class MyDataset(Dataset):
    def __init__(self, csv_file):
        self.data = pd.read_csv(csv_file)
    
    def __len__(self):
        return len(self.data)
    
    def __getitem__(self, idx):
        return torch.tensor(self.data.iloc[idx].values)

dataset = MyDataset('data.csv')
dataloader = DataLoader(dataset, batch_size=32, shuffle=True)
  1. 使用自定義的數據加載器來讀取CSV文件:
import torch

class CustomDataset(torch.utils.data.Dataset):
    def __init__(self, csv_file):
        data = pd.read_csv(csv_file)
        self.X = torch.tensor(data.iloc[:, :-1].values, dtype=torch.float32)
        self.y = torch.tensor(data.iloc[:, -1].values, dtype=torch.long)
    
    def __len__(self):
        return len(self.X)
    
    def __getitem__(self, idx):
        return self.X[idx], self.y[idx]

dataset = CustomDataset('data.csv')
dataloader = torch.utils.data.DataLoader(dataset, batch_size=32, shuffle=True)

這些是一些常用的方法,你可以根據自己的需求選擇適合的方法來讀取CSV數據集。

0
承德县| 安溪县| 台东县| 浦北县| 时尚| 吉林市| 长宁区| 阜阳市| 林甸县| 曲周县| 绥阳县| 开化县| 黄平县| 尚义县| 石首市| 泸定县| 南涧| 天镇县| 上栗县| 望城县| 龙里县| 喀什市| 绥江县| 沙湾县| 连山| 定兴县| 松阳县| 陈巴尔虎旗| 镇雄县| 昭通市| 简阳市| 尉氏县| 云梦县| 阿拉善盟| 葫芦岛市| 新河县| 平远县| 白城市| 准格尔旗| 伊吾县| 嘉荫县|