根据教程复现的代码,想通过Pytorch读取Excel中的数据并转为Tensor,结果就报错,也找不到原因
class ExcelDataset(Dataset):
def __int__(self):
train = pd.read_excel("卫星PCAtrain.xlsx",dtype=np.float32)
print(f"the shape of dataframe is {train.shape}")
x_train = train.iloc[:, :-1].values
y_train = train.iloc[:, -1].values
self.x_train = torch.from_numpy(x_train)
self.y_train = torch.from_numpy(y_train)
def __len__(self):
return len(self.y_train)
def __getitem__(self, index):
return self.x_train[index],self.y_train[index]
if name == 'main':
excel_dataset = ExcelDataset()
excel_dataloader = DataLoader(dataset=excel_dataset, batch_size=8, shuffle=True,num_workers=2)