將預測寫回數據框 pandas

Question

我正在研究 Jupyter notebook 中的一些分類模型。 我現在已經嘗試了許多不同的算法，並確定了哪一個是最佳選擇。

我需要將預測寫回原始數據幀（訓練/測試數據）嗎？ 是否有任何說明來說明這是如何完成的？

Answer 1

假設您使用的是 sklearn。

# model is the trained model. df is the features you are using for prediction
y = model.predict(df)
# just create a new column in the dataframe you predicted and assign it to the prediction
df['prediction'] = y