代码之家  ›  专栏  ›  技术社区  ›  Tomas Greif

从单个数字变量的决策树分类器中获取拆分点

  •  0
  • Tomas Greif  · 技术社区  · 8 年前

    我尝试从基于单个变量创建的决策树中获取拆分。下面的方法是正确和安全的吗?

    from sklearn.tree import DecisionTreeClassifier
    import numpy as np
    x = np.array([1,2,3,4,5,6,7])
    y = np.array([1,0,0,1,1,0,1])
    x = x.reshape(7, -1)
    
    clf = DecisionTreeClassifier()
    clf.fit(x, y)
    
    # My splits
    np.sort(clf.tree_.threshold[clf.tree_.feature == 0])
    

    我只看到 0 -2 在里面 clf.tree_.fature 我的理解是 TREE_UNDEFINED here )而其他点将是具有定义的阈值的leafs。

    推荐文章