周志华机器学习第四章例题4.4

最新推荐文章于 2022-07-05 17:28:16 发布

lzaya0000

最新推荐文章于 2022-07-05 17:28:16 发布

阅读量1.2k

点赞数

本文链接：https://blog.csdn.net/lzaya0000/article/details/104631182

版权

例题4.4的结果和书上画的图不一样，但是没问题，原因是GINI指数取最小值的时候，色泽和脐部值是一样的。

import numpy as np
dataset = np.array([[0,0,0,0,0,0,1],
                    [1,0,1,0,0,0,1],
                    [1,0,0,0,0,0,1],
                    [0,0,1,0,0,0,1],
                    [2,0,0,0,0,0,1],
                    [0,1,0,0,1,1,1],
                    [1,1,0,1,1,1,1],
                    [1,1,0,0,1,0,1],
                    [1,1,1,1,1,0,0],
                    [0,2,2,0,2,1,0],
                    [2,2,2,2,2,0,0],
                    [2,0,0,2,2,1,0],
                    [0,1,0,1,0,0,0],
                    [2,1,1,1,0,0,0],
                    [1,1,0,0,1,1,0],
                    [2