python中是否有用于非二进制决策树的库?

时间:2019-12-10 19:17:04

标签: python machine-learning scikit-learn

因此,我正在寻找一些具有非二进制决策树算法实现的库。 例如,我想使用某种决策树算法对动物园集合(https://www.kaggle.com/uciml/zoo-animal-classification/data)进行分类。 到目前为止,我已经了解了sklearn方法(DecisionTreeClassifier):

from sklearn.tree import DecisionTreeClassifier
dtree=DecisionTreeClassifier()

但是它会建立二叉树: Part of result tree

而且动物园集合具有分类数据,因此我认为最好在这里使用非二叉树(这不是重点,但是如果我错了,请更正我)。 所以我的问题是: Python中是否有任何库可以构建如下图所示的决策树: part of non binary decision tree for zoo dataset

即使没有应用任何绘图工具的输出也可以作为一个简单的字典,就像这样:

{'legs': {0: {'fins': {0.0: {'toothed': {0.0: 7.0, 1.0: 3.0}},
                       1.0: {'eggs': {0.0: 1.0, 1.0: 4.0}}}},
          2: {'hair': {0.0: 2.0, 1.0: 1.0}},
          4: {'hair': {0.0: {'aquatic': {0.0: 3.0,
                                         1.0: {'toothed': {0.0: 7.0,
                                                           1.0: 5.0}}}},
                       1.0: 1.0}},
          5: 7.0,
          6: {'aquatic': {0.0: 6.0, 1.0: 7.0}},
          8: 7.0}}

1 个答案:

答案 0 :(得分:1)

您想要的只是表示二叉树的一种更具体的方式。

您显示的树可以重写为二叉树。

如果您有分类数据,只需使用Session::get('name')