在iris数据集上决策树算法和Bagging算法的简单对比

import pandas as pd
from sklearn.model_selection import train_test_split
from bao import *
from sklearn import datasets  # 导入库
iris = datasets.load_iris()  # 导入鸢尾花数据
X = iris.data
y = iris.target

X=np.array(X)
y=np.array(y)

#print(type(x_train))

x_train, x_test, y_train, y_test = train_test_split(X, y, test_size = 0.25)
x_test, x_valid, y_test, y_valid = train_test_split(x_test, y_test, test_size = 0.5)

print(len(y_train))
print(len(y_test))
print(len(y_valid))

clf = estimators0[0]
clf.fit(x_train,y_train)
a = clf.predict(x_test)
b = accuracy_score(y_test,a)
print("决策树精度为{}".format(b))


bagging = BaggingClassifier(base_estimator=estimators0[0], n_estimators=10)
bagging.fit(x_train,y_train)
a2 = bagging.predict(x_test)
b2 = accuracy_score(y_test,a2)
print("Bagging精度为{}".format(b2))