diff --git a/sklearn/tree/export.py b/sklearn/tree/export.py index 451c0f0b1e93c..3d7f15426e50f 100644 --- a/sklearn/tree/export.py +++ b/sklearn/tree/export.py @@ -92,7 +92,7 @@ def export_graphviz(decision_tree, out_file=SENTINEL, max_depth=None, Parameters ---------- - decision_tree : decision tree classifier + decision_tree : decision tree regressor or classifier The decision tree to be exported to GraphViz. out_file : file object or string, optional (default='tree.dot')