轉(zhuǎn)載自https://blog.csdn.net/zhiyongbo/article/details/113348535
import numpy as np
from torch.utils.data import Dataset,DataLoader,TensorDataset
import math
import os
import torch
import torch.nn as nn
import torch.nn.functional as F
from torchvision import datasets, transforms
import torch.utils.data
import matplotlib.pyplot as plt
加載csv文件，直接就是矩陣
with open("./data/benign_and_male/x_train_m2.csv", encoding='utf-8') as path_x_train_m,
open("./data/benign_and_male/x_test_m2.csv", encoding='utf-8') as path_x_test_m,
open("./data/benign_and_male/y_train_m2.csv", encoding='utf-8') as path_y_train_m,
open("./data/benign_and_male/y_test_m2.csv", encoding='utf-8')as path_y_test_m:
x_train = np.loadtxt(path_x_train_m, delimiter=",")
x_test = np.loadtxt(path_x_test_m, delimiter=",")
y_train = np.loadtxt(path_y_train_m, delimiter=",")
y_test = np.loadtxt(path_y_test_m, delimiter=",")
轉(zhuǎn)換為tensor
x_train=torch.from_numpy(x_train)
y_train=torch.from_numpy(y_train)
x_test=torch.from_numpy(x_test)
y_test=torch.from_numpy(y_test)
x_train=x_train.float()
x_test=x_train.float()
y_train=y_train.long()
y_test=y_train.long()
train_dataset=TensorDataset(x_train,y_train)
test_dataset=TensorDataset(x_test,y_test)
train_loader = torch.utils.data.DataLoader(dataset=train_dataset,batch_size=batch_size, shuffle=True)
test_loader = torch.utils.data.DataLoader(dataset=test_dataset,batch_size=batch_size, shuffle=True)
版權(quán)聲明：本文為CSDN博主「飛翔的代碼人」的原創(chuàng)文章吹榴，遵循CC 4.0 BY-SA版權(quán)協(xié)議宠页，轉(zhuǎn)載請附上原文出處鏈接及本聲明。
原文鏈接：https://blog.csdn.net/zhiyongbo/article/details/113348535
