Skip to content

Commit

Permalink
update readme
Browse files Browse the repository at this point in the history
  • Loading branch information
sumedh10 committed Nov 15, 2019
1 parent 446c014 commit 4e3f759
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 2 deletions.
3 changes: 2 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -288,7 +288,8 @@ You can see what rules the tree learned by plotting this decision tree. To do th
# Create DOT data
dot_data = export_graphviz(clf, out_file=None,
feature_names=ohe_df.columns,
class_names=np.unique(y).astype('str'))
class_names=np.unique(y).astype('str'),
filled=True, rounded=True, special_characters=True)

# Draw graph
graph = graph_from_dot_data(dot_data)
Expand Down
2 changes: 1 addition & 1 deletion index.ipynb

Large diffs are not rendered by default.

Binary file modified index_files/index_11_0.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit 4e3f759

Please sign in to comment.