scikit-learn库之决策树
在scikit-learn库中决策树使用的CART算法,因此该决策树既可以解决回归问题又可以解决分类问题,即下面即将讲的DecisionTreeClassifier
和DecisionTreeRegressor
两个模型。
接下来将会讨论这两者的区别,由于是从官方文档翻译而来,翻译会略有偏颇,有兴趣的也可以去scikit-learn官方文档查看https://scikit-learn.org/stable/modules/classes.html#module-sklearn.tree
DecisionTreeClassifier
使用场景
DecisionTreeClassifier
模型即CART算法实现的决策树,通常用于解决分类问题。