Python使用XGBoost

最新推荐文章于 2024-08-20 09:50:01 发布

烂桃先生之前的账号

最新推荐文章于 2024-08-20 09:50:01 发布

阅读量317

点赞数

分类专栏：随笔

本文链接：https://blog.csdn.net/weixin_42464187/article/details/104044736

版权

本文详细介绍了如何在Python中利用XGBoost进行数据建模，包括安装库、数据预处理、模型训练、参数调优和模型评估等关键步骤，助你深入理解并掌握这一强大的机器学习库。

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

import xgboost
from xgboost import XGBClassifier
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
from sklearn.externals import joblib
import numpy as np



class XGBOOST():
    def __init__(self,model_path):
        # 加载模型
        self.model = joblib.load(model_path)

    def train():
        # 载入数据集
        dataset = np.load("model/train_data.npy")
        dataset[:,10][<