一.可以做什么
也可用于分类
二:原理(找到符合某个条件来进行分类 就是一个x furuter)
鸢尾花中petal length和petal width两个x,可以横着一次 竖着一次
防止过拟合:
优缺点
三。代码
from sklearn.tree import DecisionTreeClassifier
def iris_predict(train_sample, train_label, test_sample):
'''
实现功能:1.训练模型 2.预测
:param train_sample: 包含多条训练样本的样本集,类型为ndarray
:param train_label: 包含多条训练样本标签的标签集,类型为ndarray
:param test_sample: 包含多条测试样本的测试集,类型为ndarry
:return: test_sample对应的预测标签
'''
# ************* Begin ************#
tree_clf = DecisionTreeClassifier(splitter="random")
tree_clf = tree_clf.fit(train_sample, train_label)
y_pred = tree_clf.predict(test_sample)
return y_pred;
# ************* End **************#