四大金刚

任务1.2 随机森林、XGB、GBDT、LightGBM 四大金刚

坑大多在于安装这几个包吧,这里就不详细概述了。
直接上代码,有看不懂的评论吧。

#导入包
import pandas as pd
from sklearn.model_selection import train_test_split
import xgboost as xgb
import lightgbm as lgb
from sklearn.ensemble import RandomForestClassifier
from sklearn.ensemble import GradientBoostingRegressor 
#日常读取数据
data=pd.read_csv('data_all.csv')
x = data.drop(['status'], axis=1)
y = data['status']
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.3, random_state=2018)
#数据标准化
from sklearn import preprocessing
x_scaled = preprocessing.scale(x_train)
x_scaled_test = preprocessing.scale(x_test)
#使用模型
gbdt=GradientBoostingRegressor()
rfc=RandomForestClassifier()
gbm=lgb.LGBMRegressor()
xgb1=xgb.XGBClassifier()
#跑~
xgb1.fit(x_scaled, y_train)
gbdt.fit(x_scaled, y_train)
rfc.fit(x_scaled, y_train)
gbm.fit(x_scaled, y_train)
#成型
y_pre1=xgb1.predict(x_scaled_test)
y_pre2=gbdt.predict(x_scaled_test)
y_pre3=rfc.predict(x_scaled_test)
y_pre4=gbm.predict(x_scaled_test)
pre1 = [round(value) for value in y_pre1]
pre2 = [round(value) for value in y_pre2]
pre3 = [round(value) for value in y_pre3]
pre4 = [round(value) for value in y_pre4]
#导入F1得分
from sklearn.metrics import f1_score
#最后打印出分数
s1=f1_score(y_test,pre1)
s2=f1_score(y_test,pre2)
s3=f1_score(y_test,pre3)
s4=f1_score(y_test,pre4)
print(s1,s2,s3,s4)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值