TypeError: can‘t convert np.ndarray of type numpy.object_. The only supported types are: float64...

最新推荐文章于 2024-10-17 20:54:50 发布

上烟雨心上尘

最新推荐文章于 2024-10-17 20:54:50 发布

阅读量185

点赞数 2

分类专栏：杂谈文章标签： numpy

本文链接：https://blog.youkuaiyun.com/weixin_64632836/article/details/138174722

版权

杂谈专栏收录该内容

13 篇文章

订阅专栏

文章讲述了在使用PyTorch处理数据时，遇到将numpy数组转换为torch.tensor时的类型错误，通过astype方法将所有_features数组的类型从int转换为float，解决了训练和测试特征的创建问题，重点关注了SalePrice变量的处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

报错代码如下：

n_train = train_data.shape[0]
train_features = torch.tensor(all_features[:n_train].values, dtype=torch.float32)
test_features = torch.tensor(all_features[n_train:].values, dtype=torch.float32)
train_labels = torch.tensor(
    train_data.SalePrice.values.reshape(-1, 1), dtype=torch.float32)

解决方案

train_features = torch.tensor(all_features[:n_train].values.astype(float), dtype=torch.float32)
test_features = torch.tensor(all_features[n_train:].values.astype(float), dtype=torch.float32)