import torch
import torch.nn as nn
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import torch.nn.functional as F
from torch.utils.data import TensorDataset
from torch.utils.data import DataLoader
from sklearn.model_selection import train_test_split
data = pd.read_csv(r'D:/pycharmworkspace/ISLR-master/iris.csv')
#将数据Species列由字母转换成数字
data['Species']=pd.factorize(data.Species)[0]
#X不要第零列,不要最后一列 第一个:代表全部的行 第二个:代表列
X = data.iloc[:,1:-1].values
#Y只要特定一列
Y = data.Species.values
train_x,test_x,train_y,test_y=train_test_split(X,Y)
#将数据转换成Tensor LongTensor等价于int64
train_x = torch.from_numpy(train_x).type(torch.float32)
train_y = torch.from_numpy(train_y).type(torch.int64)
test_x = torch.from_numpy(test_x).type(torch.float32)
test_y = torch.from_numpy(test_y).type(torch.LongTensor)
#数据只有150行故batch也要小一点
batch = 8
no_of_batches = len(data)//batch
epochs = 1000
#TensorDataset()可以对tensor进行打包即合并
trai
基于pytorch的部分鸢尾花数据集多项分类
最新推荐文章于 2024-05-06 11:09:44 发布