我正在做一个决策树,我想强制算法在一个节点之后将结果分成不同的类。 问题在于,在获得条件的树中,在评估条件(X小于某个值)之后,我得到了两个相同类的结果(例如,是和是)。我想对节点进行评估的结果为“是”和“否”。 这是我得到的示例:
这是生成树和图的代码:
clf = tree.DecisionTreeclassifier(max_depth=2)
clf = clf.fit(users_data,users_target)
dot_data = tree.export_graphviz(clf,out_file=None,feature_names= feature_names,class_names= target_names,filled=True,rounded=True,special_characters=True)
graph = graphviz.Source(dot_data)
graph
我希望在节点之后找到“是”和“否”类。现在,在相应条件之后,我在最后一级获得相同的课程。
谢谢!