File tree Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Original file line number Diff line number Diff line change @@ -143,7 +143,6 @@ supported file type): ``dot -Tpdf iris.dot -o iris.pdf``.
143
143
Alternatively, if we have Python module ``pydotplus `` installed, we can generate
144
144
a PDF file (or any other supported file type) directly in Python::
145
145
146
- >>> from sklearn.externals.six import StringIO # doctest: +SKIP
147
146
>>> import pydotplus # doctest: +SKIP
148
147
>>> dot_data = tree.export_graphviz(clf, out_file=None) # doctest: +SKIP
149
148
>>> graph = pydotplus.graph_from_dot_data(dot_data) # doctest: +SKIP
@@ -155,7 +154,7 @@ using explicit variable and class names if desired. IPython notebooks can also
155
154
render these plots inline using the `Image() ` function::
156
155
157
156
>>> from IPython.display import Image # doctest: +SKIP
158
- >>> dot_data = tree.export_graphviz(clf, out_file=None, # doctest: +SKIP
157
+ >>> dot_data = tree.export_graphviz(clf, out_file=None, # doctest: +SKIP
159
158
feature_names=iris.feature_names, # doctest: +SKIP
160
159
class_names=iris.target_names, # doctest: +SKIP
161
160
filled=True, rounded=True, # doctest: +SKIP
You can’t perform that action at this time.
0 commit comments