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 中有多個列，請刪除不需要輸入訓練循環的列。