信用卡使用意图预测模型优化:提升类别1召回率
信用卡使用意图预测模型分析与优化
本研究基于信用卡使用意图预测构建模型,并对模型性能进行评估。
模型评估指标:
- ROC score
- precision
- recall
- f1-score
- support
模型性能:
- 训练集:
- ROC score: 0.7649
- accuracy: 0.80
- 类别0:
- precision: 0.81
- recall: 0.96
- f1-score: 0.88
- support: 105018
- 类别1:
- precision: 0.68
- recall: 0.27
- f1-score: 0.39
- support: 32588
- 测试集:
- ROC score: 0.7845
- accuracy: 0.79
- 类别0:
- precision: 0.81
- recall: 0.95
- f1-score: 0.87
- support: 44985
- 类别1:
- precision: 0.64
- recall: 0.28
- f1-score: 0.39
- support: 13989
结果分析:
模型在训练集和测试集上都表现出一定的预测能力,ROC score分别为0.76和0.78,accuracy均接近0.8。 然而,模型对类别1的预测能力相对较弱,召回率仅为0.27和0.28,说明模型漏掉了很多真实类别为1的样本。
优化方向:
- 数据层面:
- 增加类别1样本数量,平衡数据分布。
- 对数据进行特征工程,挖掘更有效的特征。
- 模型层面:
- 尝试使用更复杂的模型,例如集成学习模型。
- 调整模型参数,提高模型对类别1的敏感度。
- 评估指标:
- 考虑使用除ROC score和accuracy以外的指标,例如PR曲线下的面积,更全面地评估模型性能。
结论:
虽然模型初步具备一定的预测能力,但仍需进一步优化以提升对类别1的预测准确率,特别是召回率。通过数据、模型和评估指标等方面的优化,可以有效提升模型的整体性能。
原文地址: https://www.cveoy.top/t/topic/fVJT 著作权归作者所有。请勿转载和采集!