python Sklearn实现xgboost的二分类和多分类

最新推荐文章于 2024-07-05 16:14:48 发布

平步青云p

最新推荐文章于 2024-07-05 16:14:48 发布

阅读量1.6w

点赞数 8

分类专栏：学习笔记

本文链接：https://blog.csdn.net/ping550/article/details/79876298

版权

这篇博客介绍了如何利用Python的Sklearn库结合XGBoost进行二分类和多分类任务。在二分类问题中，数据集为train2.txt，而在多分类问题中，存在19种类别，包括一个正常类别和18种异常类别。文章详细阐述了数据格式及模型训练过程。

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

二分类：

train2.txt的格式如下：

import numpy as np
import pandas as pd
import sklearn
from sklearn.cross_validation import train_test_split,cross_val_score
from xgboost.sklearn import XGBClassifier
from sklearn.metrics import precision_score,roc_auc_score

min_max_scaler = sklearn.preprocessing.MinMaxScaler(feature_range=(-1,1))
resultX = []
resultY = []
with open("./train_data/train2.txt",'r') as rf:
    train_lines = rf.readlines()
    for train_line in tra