ValueError：創建 dataframe 時數據必須是一維錯誤

Question

我正在嘗試用神經網絡解決分類問題，在得到預測后，我想創建一個 pandas 數據框，其中有一列來自測試數據集，我的預測作為第二列。 但我不斷收到錯誤。 這是我的代碼：在此處輸入圖像描述

這是我的錯誤：在此處輸入圖像描述

Answer 1

重要的旁注：請花一些時間研究如何制作良好的可重現 pandas 示例，那里有關於如何更好地提出問題的很好的建議。

現在為您的錯誤：

Data must be 1-dimensional

這意味着pandas想要一個一維數組，即 [0,0,1,1,...,1] 的形式。 但是您的preds數組是二維的，即形式為[[0],[0],[1],[1],...,[1]] 。

所以你需要在這里展平preds數組：

而不是for -loops 考慮使用列表推導將您的代碼更改為以下內容：

predictions = [1 if p>0.5 else 0 for p in preds]
df = pd.DataFrame({'PassengerId': test['PassengerId'].values, 
                   'Survived': predictions})

此外，同時查看ndarray.round方法 - 也許它更適合您的用例：

predictions = preds.round()