作为一个萌新花了三天终于做出来了,普大喜奔,借鉴了机械学习实战的部分写法,把带有连续型特征值的决策树做了出来。
import math
import numpy as np
import operator
#量化的数据
dataset = np.array([[0,0,0,0,0,0,0.697,0.460,1],
[1,0,1,0,0,0,0.774,0.376,1],
[1,0,0,0,0,0,0.634,0.264,1],
[0,0,1,0,0,0,0.608,0.318,1],
[2,0,0,0,0,0,0.556,0.215,1],
[0,1,0,0,1,1,0.403,0.237,1],
[1,1,0,1,1,1,0.481,0.149,1],
[1,1,0,0,1,0,0.437,0.211,1],
[1,1,1,1,1,0,0.666,0.091,0],
[0,2,2,0,2,1,0.243,0.267,0],
[2,2,2,2,2,0,0.245,0.057,0],
[2,0,0,2,2,1,0.343,0.099,0]