尝试将数据拆分为测试集和验证集时出现张量断言错误?

2024-05-16 08:10:53 发布

您现在位置:Python中文网/ 问答频道 /正文

我有代谢物数据,我正试图建立一个神经网络模型。我得到了断言所有(张量[0].size(0)=张量。对于张量中的张量,size(0))断言错误,我不知道该怎么做才能使它正常工作

问题就发生在这里

#Split data into training and validation 70 / 30
data = torch.utils.data.TensorDataset(torch.Tensor(np.array(log_metabol)),
                                                torch.Tensor(labels))
train, val = torch.utils.data.random_split(data, [int(0.7 * len(data)), 
                                                              len(data) - int(0.7 * len(data))])
 

这个问题似乎起源于这个街区的第一排。我该怎么做


Tags: 数据模型datasizelen错误utils神经网络