在信用卡 使用意图预测中运用lightgbm模型对已有数据进行训练测试以下是代码请详细说明该代码的作用import lightgbm as lgblg = lgbLGBMClassifierparams = boosting_typegdbt dart rf max_depthsp_randint-1 20 learning_rate01 02030405 n_estimator
该代码的作用是使用lightgbm模型对信用卡使用意图进行预测,包括以下步骤:
- 定义了一个LGBMClassifier对象lg。
- 定义了一个参数字典params,包含了boosting_type、max_depth、learning_rate、n_estimators等参数,用于进行随机搜索。
- 创建一个RandomizedSearchCV对象rscv,使用param_distributions参数指定了参数搜索的范围,cv参数指定了交叉验证的次数,scoring参数指定了评分函数,n_iter参数指定了搜索的次数,n_jobs参数指定了并行运算的数量。
- 对数据进行拟合训练,得到最佳参数rscv.best_params_。
- 创建一个新的LGBMClassifier对象lg,使用最佳参数进行训练。
- 对训练集进行预测,计算ROC得分、分类报告和混淆矩阵。
- 对测试集进行预测,计算ROC得分、分类报告和混淆矩阵。
原文地址: https://www.cveoy.top/t/topic/fQc3 著作权归作者所有。请勿转载和采集!