书籍信息
Hands-On Machine Learning with Scikit-Learn and Tensorflow
出版社: O’Reilly Media, Inc, USA
平装: 566页
语种: 英语
ISBN: 1491962291
条形码: 9781491962299
商品尺寸: 18 x 2.9 x 23.3 cm
ASIN: 1491962291
系列博文为书籍中文翻译
代码以及数据下载:https://github.com/ageron/handson-ml
此为6.1得到的决策树
决策树也能够估计实例归属特定类别的概率,基本思想是返回叶子结点中特定类别的比例。例如,如果petal length=5厘米,petal width=1.5厘米,那么实例对应的叶子结点是左下角的结点,类别的比例是
setosa:0%(0/54)
versicolor:90.7%(49/54)
virginica:9.3%(5/54)
类别的比例即是类别的概率。预测的类别是概率最高的类别,即versicolor。我们可以通过代码检验我们的结果。
tree_clf.predict_proba([[5, 1.5]])
# output
# array([[ 0. , 0.90740741, 0.09259259]])
tree_clf.predict([[5, 1.5]])
# output
# array([1])