sklearn之adaboost

# -*- coding: utf-8 -*-
import pandas
import numpy as np
import matplotlib.pyplot as plt

from sklearn.ensemble import AdaBoostClassifier
from sklearn.tree import DecisionTreeClassifier

from sklearn.metrics import accuracy_score
from sklearn.model_selection import cross_val_score
#cross_val_score的参数cv表示几折交叉验证，scoring默认是accuracy
from sklearn.model_selection import cross_val_predict
from sklearn.model_selection import train_test_split
from sklearn.model_selection import learning_curve

from sklearn.datasets import load_digits
#载入数据
dataset=load_digits()
x=dataset['data']
y=dataset['target']

plt.imshow(x[4].reshape(8,8))
#plt.show()

reg_ada=AdaBoostClassifier(DecisionTreeClassifier(max_depth=1))
scores_ada=cross_val_score(reg_ada,x,y,cv=6)
print(scores_ada.mean())

score=[]
for depth in [1,2,10]:
    reg_ada=AdaBoostClassifier(DecisionTreeClassifier(max_depth=depth))
    scores_ada=cross_val_score(reg_ada,x,y,cv=6)
    score.append(scores_ada.mean())
    print(scores_ada