如何从 XGBoost 模型中导出预测？答案

【问题标题】：How do I export the predictions from XGBoost model?如何从 XGBoost 模型中导出预测？
【发布时间】：2020-07-25 22:36:51
【问题描述】：

我已经训练了我的 XGBoost 分类器模型并在测试数据上取得了结果。我能够计算混淆矩阵。现在我需要支持在资产级别导出我的测试预测。我想要资产级别的输出 123 ID - 1 234 身份证 - 0 等等。

甚至概率会代替 1 和 0，这样我就可以找出我的模型的弱点并深入挖掘模型的性能。

如何在 python 中实现这一点？你能帮忙吗？

【问题讨论】：

标签： python output export-to-csv prediction xgboost

【解决方案1】：

你可以这样使用：

# import pandas as pd
with pd.option_context('display.max_rows', None, 'display.max_columns', None):
    preds = predict(xgb.model, test_x)
    patient_id = test_x['patient_ID']
    dframe = pd.DataFrame(patient_id)
    dframe['Response'] = list(test_y)
    dframe['Prediction'] = list(preds)
    print(dframe)

【讨论】：