91超碰碰碰碰久久久久久综合_超碰av人澡人澡人澡人澡人掠_国产黄大片在线观看画质优化_txt小说免费全本

溫馨提示×

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
黔西| 西峡县| 潜江市| 肃北| 天气| 杭锦后旗| 阳高县| 南充市| 南召县| 玉溪市| 克山县| 孙吴县| 丰台区| 若羌县| 岳阳市| 高雄市| 怀集县| 上高县| 黑水县| 特克斯县| 禹州市| 江津市| 那曲县| 奎屯市| 忻州市| 本溪| 沧源| 青海省| 南江县| 新兴县| 东至县| 林西县| 临夏县| 防城港市| 临沭县| 莱州市| 公主岭市| 科技| 新安县| 镇坪县| 扶沟县|