Open saljamil opened 10 months ago
Page 27 of the book - had to change the code to this: fig, ax = plt.subplots(figsize=(8, 4)) tree.plot_tree(stump, feature_names=['value'], filled=True, classnames=list(stump.classes), ax=ax)
with class_names param needing a list
Page 27 of the book - had to change the code to this: fig, ax = plt.subplots(figsize=(8, 4)) tree.plot_tree(stump, feature_names=['value'], filled=True, classnames=list(stump.classes), ax=ax)
with class_names param needing a list