def getTreeSplits(dt):
'''
????????????????
dt: ????????????sklearn.tree.DecisionTreeClassifier?
???????None????????????
'''
cut=dt.tree_.threshold[np.where(dt.tree_.children_left>-1)]
if cut.shape[0]==0:
return None
return np.sort(cut)
评论列表
文章目录