xgboost对密西西比数据集csv文件进行预测

优游涵泳斋

已于 2024-01-07 12:55:58 修改

阅读量539

点赞数 8

文章标签：公式序号 python 神经网络

于 2024-01-06 10:52:54 首次发布

本文链接：https://blog.csdn.net/weixin_48433993/article/details/135423092

版权

代码：

# 导入需要的库
from sklearn.preprocessing import LabelEncoder
import matplotlib.pyplot as plt
import pandas as pd
import xgboost as xgb
from sklearn.model_selection import train_test_split
from sklearn.metrics import confusion_matrix, classification_report, accuracy_score
import numpy as np
import math

#获取数据转为numpy数组
filename=pd.read_csv("C:/Users/86170/Desktop/exel/testdata.csv")
filename1=pd.read_csv("C:/Users/86170/Desktop/exel/test.csv")
sheet1=np.array(filename)
sheet2=np.array(filename1)

#分成特征与标签
le = LabelEncoder()
#label_mapping = {0: 'AD', 1: 'CN', 2: 'EMCI', 3: 'LMCI', 4: 'SMC'}
X=np.array(sheet1[1:10001,0:26])
Y=np.array(sheet1[1:10001,26:27])

new_X=np.array(sheet2[1:10001,0:26])
X_train, X_test, y_train, y_test = train_test_split(X, Y, test_size=0.2, random_state=42)
y_train=le.fit_transform(y_train)
y_test=le.fit_transform(y_test)
# 创建B