机器学习决策树2个经典案例

机器学习决策树import operator
from math import log
import matplotlib.pyplot as plt

def createDataSet():
    dataSet = [[ '杂食动物', '是', '否', '否', '是'],
               [ '杂食动物', '是', '否', '否', '是'],
               [ '肉食动物', '是', '否', '否', '是'],
               [ '肉食动物', '否', '否', '是', '否'],
               [ '肉食动物', '否', '是', '否', '否'],
               [ '肉食动物', '否', '否', '否', '否'],
               [ '杂食动物', '是', '否', '是', '是'],
               [ '草食动物', '是', '否', '否', '是'],
               [ '杂食动物', '否', '否', '是', '否'],
               [ '肉食动物', '否', '是', '否', '否'],
               [ '肉食动物', '是', '是', '否', '是'],
               [ '肉食动物', '否', '否', '否', '是'],
               [ '草食动物', '是', '否', '否', '是'],
               [ '肉食动物', '否', '否', '否', '否']]
    labels = ['饮食习惯', '胎生动物', '水生动物', '会飞']
    return dataSet, labels