我有一个 if-else 规则决策树文件，并且想要可视化它，我该怎么做?

Question

我使用 C4.5 算法生成了一个决策树，该树作为 if-else 规则生成并存储在rules.py中。

现在我想在 rules.py 中可视化决策树。我看过一些图形可视化的教程，但我无法可视化它。

以下是rules.py文件：

def findDecision(obj): #obj[0]: Semester, obj[1]: Gender, obj[2]: Friend_time, obj[3]: Faculty_visit, obj[4]: avg_attendance, obj[5]: attentive_class, obj[6]: participation_corricular, obj[7]: study_time, obj[8]: social_time, obj[9]: cgpa, obj[10]: extra_crse, obj[11]: health, obj[12]: sleep, obj[13]: go_out_frnd, obj[14]: lib_time
if obj[9]>2:
if obj[14]>2:
if obj[12]>1:
if obj[3]>1:
if obj[7]>1:
return 5
elif obj[7]<=1:
return 4.5
else:
return 4.5
elif obj[3]<=1:
return 5.666666666666667
else:
return 5.666666666666667
elif obj[12]<=1:
return 6.5
else:
return 6.5
elif obj[14]<=2:
if obj[11]>1:
if obj[8]>1:
if obj[2]>2:
return 6.0
elif obj[2]<=2:
return 7
else:
return 7.0
elif obj[8]<=1:
return 6
else:
return 6.0
elif obj[11]<=1:
return 8.0
else:
return 8.0
else:
return 6.7272727272727275
elif obj[9]<=2:
if obj[7]>1:
if obj[1]>1:
return 8
elif obj[1]<=1:
return 9
else:
return 9.0
elif obj[7]<=1:
return 7.0
else:
return 7.0
else:
return 8.125

那么请帮我如何可视化吗？

Answer 1

你可以使用 pythontutor 来可视化代码。