Skip to content

Commit 9c8541f

Browse files
lucianoviolapaulha
authored andcommitted
added tree of type "regressor" to the docstring of exportviz (scikit-learn#9530)
1 parent b41c3c7 commit 9c8541f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

sklearn/tree/export.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ def export_graphviz(decision_tree, out_file=SENTINEL, max_depth=None,
9292
9393
Parameters
9494
----------
95-
decision_tree : decision tree classifier
95+
decision_tree : decision tree regressor or classifier
9696
The decision tree to be exported to GraphViz.
9797
9898
out_file : file object or string, optional (default='tree.dot')

0 commit comments

Comments
 (0)