import pandas as pd
from sklearn import datasets
from sklearn.preprocessing import LabelEncoder
from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
导入数据
X, y = datasets.load_wine(return_X_y=True)
数据预处理
le = LabelEncoder()# 把label转换为0和1
y = le.fit_transform(y)# 训练集和测试集比例为7:3
X_train, X_test, y_train, y_test = train_test_split(
X, y, test_size=0.30, random_state=1)