ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

特征筛选伪代码

2022-04-17 23:33:22  阅读:177  来源: 互联网

标签:feas name 特征 代码 xgb date 筛选 copy mse


xgb=XGB(trial=False)
trainded_model=xgb.train(trainset,date_name='date',label_name='target')
pred,gt=xgb.test(testset,trainded_model,date_name='date',label_name='target')
base_mse=evaluation(pred.values ,gt.values ,method='mse')
print('base_mse:',base_mse)

feas=['ws10','wd10','ws30','wd30','ws50','wd50','ws70','wd70','TEMP','HUMID']

while True:
l1=len(feas)
for fea in feas:
xgb=XGB(trial=False)
trainset_copy=trainset.copy()
testset_copy=testset.copy()
del trainset_copy[fea]
del testset_copy[fea]
trainded_model=xgb.train(trainset_copy,date_name='date',label_name='target')
pred,gt=xgb.test(testset_copy,trainded_model,date_name='date',label_name='target')
mse=evaluation(pred.values ,gt.values ,method='mse')
print('mse:',mse)
if mse<base_mse:
feas.remove(fea)###当找到能降低mse的无用特征时,永久删除该特征,然后break,开启下一段循环
break
if len(feas)==l1:###当找不到无用特征,feas数量收敛时,断开while True
print(feas)
break

标签:feas,name,特征,代码,xgb,date,筛选,copy,mse
来源: https://www.cnblogs.com/chentianyu/p/16157867.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有