决策树分析例题经典案例_机器学习决策树2个经典案例

【实例简介】

机器学习决策树

【实例截图】

【核心代码】

import operator

from math import log

import matplotlib.pyplot as plt

def createDataSet():

dataSet = [[ '杂食动物', '是', '否', '否', '是'],

[ '杂食动物', '是', '否', '否', '是'],

[ '肉食动物', '是', '否', '否', '是'],

[ '肉食动物', '否', '否', '是', '否'],

[ '肉食动物', '否', '是', '否', '否'],

[ '肉食动物', '否', '否', '否', '否'],

[ '杂食动物', '是', '否', '是', '是'],

[ '草食动物', '是', '否', '否', '是'],

[ '杂食动物', '否', '否', '是', '否'],

[ '肉食动物', '否', '是', '否', '否'],

[ '肉食动物', '是', '是', '否', '是'],

[ '肉食动物', '否', '否', '否', '是'],

[ '草食动物', '是', '否', '否', '是'],

[ '肉食动物', '否', '否', '否', '否']]

labels = ['饮食习惯', '胎生动物', '水生动物', '会飞']

return dataSet, labels