ValueError：pytorch 中的尺寸“str”太多 python

Question

我不断收到错误：

ValueError：太多维度“str”

我附上我的 colab 笔记本来查看它。 还没有在网上找到任何可以帮助我解决问题的东西。

链接： https://colab.research.google.com/drive/1ikol2D8mmiIPKhNHbcFlTfVpuU_Gf9BZ?usp=sharing

Answer 1

我在我的 Jupyter 笔记本中也看到了这个错误。 我可以使用以下简单代码重现它：

Input:
tensor(['a'])

Output: 
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-55-3bdb0dbfafc2> in <module>
----> 1 tensor(['a'])
...(stacktrace)...

ValueError: too many dimensions 'str'

显然 PyTorch 张量与 Numpy arrays 不同，因为它们不适用于字符串，仅适用于整数、浮点数和布尔值。

上述错误表明传递给张量的字符串太多（即即使一个字符串也太多）。 当我将代码更改为以下内容时，它可以正常工作：

Input:
tensor([1])

Output: 
tensor([1])

Answer 2

我还没有检查你的笔记本，但我刚刚解决了同样的错误。 只需仔细检查所有训练数据集和标签是否都转换为数值或张量。 如果 dataframe 中有多个列，请删除不需要输入训练循环的列。