[MRG] Improve the error message of export_graphviz if a not-fitted decision tree is provided (scikit-learn#8776)

Author: qinhanmin2014

sklearn/tree/export.py

@@ -14,6 +14,7 @@
 import warnings
 
 from ..externals import six
+from ..utils.validation import check_is_fitted
 
 from . import _criterion
 from . import _tree
@@ -377,6 +378,7 @@ def recurse(tree, node_id, criterion, parent=None, depth=0):
                 # Add edge to parent
                 out_file.write('%d -> %d ;\n' % (parent, node_id))
 
+    check_is_fitted(decision_tree, 'tree_')
     own_file = False
     return_string = False
     try:

sklearn/tree/tests/test_export.py

@@ -9,6 +9,7 @@
 from sklearn.tree import export_graphviz
 from sklearn.externals.six import StringIO
 from sklearn.utils.testing import assert_in, assert_equal, assert_raises
+from sklearn.exceptions import NotFittedError
 
 # toy sample
 X = [[-2, -1], [-1, -1], [-1, -2], [1, 1], [1, 2], [2, 1]]
@@ -210,6 +211,11 @@ def test_graphviz_toy():
 def test_graphviz_errors():
     # Check for errors of export_graphviz
     clf = DecisionTreeClassifier(max_depth=3, min_samples_split=2)
+
+    # Check not-fitted decision tree error
+    out = StringIO()
+    assert_raises(NotFittedError, export_graphviz, clf, out)
+
     clf.fit(X, y)
 
     # Check feature_names error