torch数据集导入

import numpy as np
import pandas as pd
import torch
from torch.utils.data import Dataset, DataLoader, TensorDataset

文件读取

将数据从文件中读取出来并转化为tensor类，即便进行后续的处理

使用pandas读取

data = pd.read_csv('train.csv', index_col=0)
x_train = torch.from_numpy(data.iloc[:, :-1].values)
x_labels = torch.from_numpy(data.iloc[:, [-1]].values)

使用numpy读取

注意 numpy只能读取纯数据，如果有表头会报错

data = np.loadtxt('train.csv', dtype=np.float64, delimiter=',')
x_train = data[:, :-1