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.
2 出错代码
X = torch.from_numpy(X_data).type(torch.float32)
3 出错原因
在numpy转为tensor时出现错误,原因是原数据集中存在不是数值的数据类型。在pycharm中可以用print(data.info())语句查看数据集每列的数据类型。
4 解决方案
此次用的数据集的part列是Object类型的,使用pd.get_dummies(data.part)语句将文本数值化,转换为one-hot编码,然后使用join()函数将one-hot列加入到原数据列中。