机器学习之决策树算法

目录

---

---

一、数据集准备

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
import warnings
warnings.filterwarnings('ignore')
import  ssl
ssl._create_default_https_context = ssl._create_unverified_context

二、应用

1.设置深度为1时

clf=tree.DecisionTreeClassifier(max_depth=1)
clf.fit(X_train,y_train)
cmap_light=ListedColormap(['#FFAAAA','#AAFFAA','#AAAAFF'])
cmap_bold=ListedColormap(['#FF0000','#00FF00','#0000FF'])
x_min,x_max=X[:,0].min()-0.5,X[:,0].max()+0.5#导入数据集拆分工具\n",
y_min,y_max=X[:,1].min()-0.5,X[:,1].max()+0.5
xx,yy=np.meshgrid(np.arange(x_min,x_max, .02),
                np.arange