user29687883
user29687883

Reputation: 1

Draw a decision tree while hiding the values ​of the "value" row

I want to simplify the decision tree output and hide the values ​​in the "value" field.Below is the code I am using

enter image description here

  fig, ax = plt.subplots(figsize=(10, 10)) 
  plt.figure(figsize=(210,10))
  plot_tree(m,
  feature_names=X.columns, 
  filled=True, 
  impurity=True, 
  rounded=True,
  proportion = True,
  max_depth = 3)

  plt.show()

Upvotes: 0

Views: 25

Answers (0)

Related Questions