【天池】贷款违规风险预测task4

最新推荐文章于 2024-04-27 00:34:47 发布

爱学习的小崽子

最新推荐文章于 2024-04-27 00:34:47 发布

阅读量401

点赞数

分类专栏：数据挖掘 python 文章标签： python 机器学习

本文链接：https://blog.csdn.net/qq_38444706/article/details/108783629

版权

这篇博客介绍了一个简单的LightGBM模型用于天池贷款违约风险预测任务。作者首先加载数据，然后对非数值列进行编码，接着利用KFold进行交叉验证，评估模型在验证集上的AUC。最后，模型在整个训练集上重新训练，并预测测试集，生成提交结果文件。

摘要由CSDN通过智能技术生成

天池贷款违规风险预测task4

最近时间紧张，仅简单跑一个验证模型，后续还需完善。

# lightGBM方法简单建模待调参完善
# 本地验证 0.7315930724616382


# lightGBM方法
import pandas as pd
import numpy as np
from category_encoders.target_encoder import TargetEncoder
from sklearn.model_selection import KFold
from sklearn.metrics import auc, roc_curve
from lightgbm import LGBMRegressor

# 导入数据
train = pd.read_csv('train.csv', index_col='id')
test = pd.read_csv('testA.csv', index_col='id')
target = train.pop('isDefault')