From 505f52db86bf4c1293ac2ed8f2daf80c79f652c8 Mon Sep 17 00:00:00 2001 From: Luciano Viola Date: Fri, 11 Aug 2017 15:08:16 -0300 Subject: [PATCH] added tree of type "regressor" to the docstring of exportviz --- sklearn/tree/export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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')