【pytorch】zh-v2.d2l数据预处理报错
温馨提示：
本文最后更新于 2024年03月30日，已超过 30 天没有更新。若文章内的图片失效（无法正常加载），请留言反馈或直接联系我。
背景
动手学深度学习pytorch：2.2.2处理缺失值报错
inputs, outputs = data.iloc[:, 0:2], data.iloc[:, 2]
inputs = inputs.fillna(inputs.mean())
print(inputs)
报错如下
TypeError: can only concatenate str (not "int") to str
解决方案
我们看出报错是因为第二列数据是字符串不是数字，修改代码即可。
inputs, outputs = data.iloc[:, 0:2], data.iloc[:, 2]
inputs = inputs.fillna(inputs.select_dtypes(include='number').mean())
print(inputs)
成功输出
NumRooms Alley
0 3.0 Pave
1 2.0 NaN
2 4.0 NaN
3 3.0 NaN
接着报错
inputs = pd.get_dummies(inputs, dummy_na=True)
print(inputs)
输出
NumRooms Alley_Pave Alley_nan
0 3.0 True False
1 2.0 False True
2 4.0 False True
3 3.0 False True
并不是教程中的0和1，而是布尔值。
这会引起后续转成张量的报错
TypeError: can't convert np.ndarray of type numpy.object_. The only supported types are: float64, float32, float16, complex64, complex128, int64, int32, int16, int8, uint8, and bool.
原因
get_dummies函数在pandas1.6.0版本之前返回numpy.uint8，无符号八位整数。
在1.6.0版本开始更改为返回numpy.bool_，numpy布尔值。
解决方案
添加dtype，完成输出0和1，且后续不报错
inputs = pd.get_dummies(inputs, dummy_na=True, dtype = 'uint8')
print(inputs)
正文到此结束
- 本文标签： pytorch
- 本文链接： https://blog.voiceclouds.cn/article/104
- 版权声明： 本文由广州海琳信息技术有限公司原创发布，转载请遵循《署名-非商业性使用-相同方式共享 4.0 国际 (CC BY-NC-SA 4.0)》许可协议授权
热门推荐
关于我
海琳博客
海琳博客为信息化时代而生
