【异常】TypeError: can't convert np.ndarray of type numpy.object_. The only supported types are: double,

最新推荐文章于 2024-08-01 18:22:21 发布

轮子去哪儿了

最新推荐文章于 2024-08-01 18:22:21 发布

阅读量3.4w

点赞数 30

分类专栏： Pytorch 学习笔记

本文链接：https://blog.csdn.net/weixin_42419002/article/details/100095302

版权

当尝试将np.ndarray对象转换时遇到错误：'TypeError: can't convert np.ndarray of type numpy.object_. The only supported types are: double, float, float16, int64, int32, and uint8.'。问题源于数据集生成过程中的数据形状不正确。在getAllData.py中，由于筛选操作导致元素长度不一致，修正方法是在创建二维list时确保所有元素长度相同。通过限制X的长度为100，解决了输入数据长度不一的问题，从而避免了torch数据加载时的错误。" 18733817,351306,Linux Xen 恢复virbr0网络接口,"['Linux系统管理', '虚拟化', '网络配置', 'virsh命令']

摘要由CSDN通过智能技术生成

1. 问题：

TypeError: can’t convert np.ndarray of type numpy.object_. The only supported types are: double, float, float16, int64, int32, and uint8.

2. 异常代码行：

train_x = torch.from_numpy(train_x).float().unsqueeze(1)  # numpy 转成 torch 类型

3. 相关代码：

train.py

from getData import getAllData
from sklearn.model_selection import train_test_split

data = getAllData()
X, Y = data.XY()
print(type(X))  # <class 'list'>
print(np.shape