xgboost代码示例

最新推荐文章于 2024-08-04 22:13:51 发布

bboyzqh

最新推荐文章于 2024-08-04 22:13:51 发布

阅读量3.5k

点赞数

分类专栏：算法机器学习文章标签： xgboost 代码

本文链接：https://blog.csdn.net/zhuqiuhui/article/details/72584376

版权

机器学习同时被 2 个专栏收录

31 篇文章 19 订阅 ¥29.90 ¥99.00

订阅专栏

算法

23 篇文章 1 订阅

订阅专栏

本文提供了一段可运行的xgboost代码示例，经过重新调试以适应最新版本。虽然数据需要替换为自定义数据集，但展示了xgboost的强大性能。欢迎深入探讨。

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

之前写过很久了，怕新更新的xgboost不再适用，重新调试了一下代码，可运行，但数据得换成自己的，xgboost，都应该知道它的威力了，这里不再多说，欢迎一起讨论！

# coding=utf-8
import pandas as pd
import xgboost as xgb
from sklearn import metrics
import matplotlib.pylab as plt
from sklearn.model_selection import train_test_split


# 训练模型并预测出结果
def train_model(train_xy, test_xy, random_seed):
    test_ID = test_xy.ID 
    test_y = test_xy.Kind
    test_xy = test_xy.drop(['ID'], axis=1)     # 去掉ID
    test_xy = test_xy.drop(['Kind'], axis=1)   # 去掉类标
    dtest = xgb.DMatrix(test_xy)


    train_y = train_xy.Kind
    train_xy = train_xy.drop(['ID'], axis=1)
    # train_xy = train_xy.dro

了解本专栏