最后一篇,主要介绍一下目前比较火的机器学习算法库Sklearn库实现决策树算法及可视化
Sklearn决策树算法库包含:
(1)分类决策树DecisionTreeClassifier
(2)回归决策树DecisionTreeRegressor
1. DecisionTreeClassifier() 示例
import pandas as pd
from sklearn.tree import DecisionTreeClassifier
from sklearn.tree import export_graphviz
import graphviz
# 统计学习方法书上题目5.1示例数据
def create_data():
datasets = [
['青年', '否', '否', '一般', '否'],
['青年', '否', '否', '好', '否'],
['青年', '是', '否', '好', '是'],
['青年', '是', '是', '一般', '是'],
['青年', '否', '否', '一般', '否'],
['中年', '否', '否', '一般', '否'],
['中年', '否', '否',