Hi,
I just begin with scikit-learn and would like to visualize a classification
tree with class names displayed in the leaves as shown in the SCIKITLEARN.TREE
documentation http://scikit-learn.org/stable/modules/tree.html where we find
class=’virginica’ etc…
I made a tree providing a 2D array X (n1 samples , n2 features) and 1D array Y
(n1 corresponding classes ) such that Y(i) is the class of the sample X(i, …)
After that I have correct predictions using predict()
Then I use the function
export_graphviz(clf, out_file=dot_data,feature_names=FEATURES)
with FEATURES being the array of my n2 features names in the same order as in X
I obtain the tree .png but can’t find a way to have the correct class names in
the leaves…
In export_graphviz() should I use the class_names optional parameter and how ?
Thanks for any help
Gregory, Toulouse FRANCE
_______________________________________________
scikit-learn mailing list
[email protected]
https://mail.python.org/mailman/listinfo/scikit-learn
