ó ‡ˆ\c@sÁdZddlmZddlZddlmZddlmZddl m Z dd l m Z d „Z d e fd „ƒYZeƒZddddd eeeeeeeedd„ZdS(s: This module defines export functions for decision trees. iÿÿÿÿ(tIntegralNi(tsix(tcheck_is_fittedi(t _criterion(t_treec Cs6g}d \}}||}||}x tjddd|ƒjtƒD]å}|d}|dt|ddƒ}||d f||d fd ||fd ||f|d |f|d |f||d fg} | t|ƒ\} } } td | |ƒtd | |ƒtd | |ƒg} |j| ƒqIW|S( s Generate n colors with equally spaced hues. Parameters ---------- n : int The number of colors required. Returns ------- color_list : list, length n List of n tuples of form (R, G, B) being the components of each color. gè?gÍÌÌÌÌÌì?iig€v@gN@iiiiÿ(gè?gÍÌÌÌÌÌì?(tnptarangetastypetinttabstappend( tnt color_listtstvtctmthth_bartxtrgbtrtgtb((s2lib/python2.7/site-packages/sklearn/tree/export.pyt _color_brews(    )       tSentinelcBseZd„ZRS(cCsdS(Ns "tree.dot"((tself((s2lib/python2.7/site-packages/sklearn/tree/export.pyt__repr__Ds(t__name__t __module__R(((s2lib/python2.7/site-packages/sklearn/tree/export.pyRCstallic s‡fd†‰‡‡‡‡‡‡ ‡ ‡fd†‰ d d‡‡‡‡‡ ‡ ‡ ‡‡f d†‰t|dƒt}t}zƒtˆ tjƒrÇtjr¯tˆ dddƒ‰ ntˆ d ƒ‰ t}nˆ d krèt}tj ƒ‰ ntˆ t ƒrˆ dkr9t d j ˆ ƒƒ‚q9nt d j t ˆ ƒƒƒ‚ˆd k rtˆƒ|jkrt d tˆƒ|jfƒ‚qnigd 6‰ id d6‰ˆ jdƒˆ jdƒg}ˆrÏ|jdƒn| rå|jdƒnt|ƒdkrˆ jddj|ƒƒn| r*ˆ jdƒnˆ jdƒ|rMˆ jdƒn| rcˆ jdƒnˆryˆ jdƒnˆ|jdd|jƒ|rÞxCtˆ ƒD]2}ˆ jddjd„ˆ |Dƒƒdƒq¥Wnˆ jdƒ|rûˆ jƒSWd |rˆ jƒnXd S(!s%Export a decision tree in DOT format. This function generates a GraphViz representation of the decision tree, which is then written into `out_file`. Once exported, graphical renderings can be generated using, for example:: $ dot -Tps tree.dot -o tree.ps (PostScript format) $ dot -Tpng tree.dot -o tree.png (PNG format) The sample counts that are shown are weighted with any sample_weights that might be present. Read more in the :ref:`User Guide `. Parameters ---------- decision_tree : decision tree regressor or classifier The decision tree to be exported to GraphViz. out_file : file object or string, optional (default=None) Handle or name of the output file. If ``None``, the result is returned as a string. .. versionchanged:: 0.20 Default of out_file changed from "tree.dot" to None. max_depth : int, optional (default=None) The maximum depth of the representation. If None, the tree is fully generated. feature_names : list of strings, optional (default=None) Names of each of the features. class_names : list of strings, bool or None, optional (default=None) Names of each of the target classes in ascending numerical order. Only relevant for classification and not supported for multi-output. If ``True``, shows a symbolic representation of the class name. label : {'all', 'root', 'none'}, optional (default='all') Whether to show informative labels for impurity, etc. Options include 'all' to show at every node, 'root' to show only at the top root node, or 'none' to not show at any node. filled : bool, optional (default=False) When set to ``True``, paint nodes to indicate majority class for classification, extremity of values for regression, or purity of node for multi-output. leaves_parallel : bool, optional (default=False) When set to ``True``, draw all leaf nodes at the bottom of the tree. impurity : bool, optional (default=True) When set to ``True``, show the impurity at each node. node_ids : bool, optional (default=False) When set to ``True``, show the ID number on each node. proportion : bool, optional (default=False) When set to ``True``, change the display of 'values' and/or 'samples' to be proportions and percentages respectively. rotate : bool, optional (default=False) When set to ``True``, orient tree left to right rather than top-down. rounded : bool, optional (default=False) When set to ``True``, draw node boxes with rounded corners and use Helvetica fonts instead of Times-Roman. special_characters : bool, optional (default=False) When set to ``False``, ignore special characters for PostScript compatibility. precision : int, optional (default=3) Number of digits of precision for floating point in the values of impurity, threshold and value attributes of each node. Returns ------- dot_data : string String representation of the input tree in GraphViz dot format. Only returned if ``out_file`` is None. .. versionadded:: 0.18 Examples -------- >>> from sklearn.datasets import load_iris >>> from sklearn import tree >>> clf = tree.DecisionTreeClassifier() >>> iris = load_iris() >>> clf = clf.fit(iris.data, iris.target) >>> tree.export_graphviz(clf, ... out_file='tree.dot') # doctest: +SKIP cstˆddkr‘tˆdtj|ƒƒ}t|dtƒ}t|ƒdkrZd}qåttjd|d|dd|ddƒƒ}nTtˆddƒ}ttjd|ˆddˆddˆdddƒƒ}|j |ƒgt dƒD]}t |ƒ^qÿ}|j dd d d d d gƒg|D] }||d||d^q=}ddj |ƒS(NtboundsRtreverseiiiÿi taRRtdtetfit#t(tNonetlistRtargmaxtsortedtTruetlenRtroundR trangetstrtextendtjoin(tvaluetcolort sorted_valuestalphatit hex_codesR(tcolors(s2lib/python2.7/site-packages/sklearn/tree/export.pyt get_color²s"    %-c sÀ|jdkr/|j|ddd…f}n |j|}ˆdkrT|dkp]ˆdk}ˆr‡ddddd d g}d }nd d ddddg}d}ˆrá|r¾|d7}n||dt|ƒ|d7}n|j|tjkrnˆdk rˆ|j|}n"d|d|j||df}|d||dt|j |ˆƒ|df7}nˆrìt |t j ƒrd}nt |t jƒsªd}n|rÁ|d|7}n|tt|j|ˆƒƒ|d7}n|rÿ|d7}nˆrOd|j|t|jdƒ}|tt|dƒƒd|d7}n|t|j|ƒ|d7}ˆr›|jddkr›||j|}n|r®|d7}n|jddkrÖtj|ˆƒ} nfˆrñtj|ˆƒ} nKtjtjtj|dƒdƒƒr*|jtƒ} ntj|ˆƒ} t| jdƒƒjd d!ƒ} | jd"d#ƒjd!d$ƒ} |jddkr¾|jdkr¾| jd d$ƒjdd$ƒ} n| jd%|dƒ} || |d7}ˆdk rz|jddkrz|jdkrz|r'|d&7}nˆtk rIˆtj|ƒ} n$d'|dtj|ƒ|df} || 7}n|d(dkr—|d( }n|d)d kr´|d) }n||d*S(+NiitrootRs#sss≤s
t>t %di-iÓÿÿÿgà?iþÿÿÿs [labeldistance=2.5, labelangle=s%d, headlabel="True"]s%d, headlabel="False"]s ; RWtparenttdepths%d [label="(...)"s, fillcolor="#C0C0C0"s %d -> %d ; (RRGt ValueErrorRFtchildren_rightR'R R/twriteRRORERtminRAtmaxR,tuniqueR2RPtarray( RURVRWRgRht left_childt right_childtnode_valtangles( R8tfilledR9t max_depthRetout_filetrankstrecursetrotate(s2lib/python2.7/site-packages/sklearn/tree/export.pyRx<sh    #   !    ttree_twtencodingsutf-8twbs<'precision' should be greater or equal to 0. Got {} instead.s1'precision' should be an integer. Got {} instead.sALength of feature_names, %d does not match number of features, %dRfRsdigraph Tree { snode [shape=boxRttroundeds, style="%s", color="black"s, s, fontname=helveticas] ; s,graph [ranksep=equally, splines=polyline] ; sedge [fontname=helvetica] ; s rankdir=LR ; RWs {rank=same ; s; css|] }|VqdS(N((t.0R((s2lib/python2.7/site-packages/sklearn/tree/export.pys Ñss} ; t}N(R'RtFalseRJRRLtPY3topenR+tStringIORRitformatttypeR,t n_features_RkR R1RzRWR*tgetvaluetclose(t decision_treeRvRuR_R^R`Rttleaves_parallelRARaRcRyR~RdRbtown_filet return_stringtrounded_filledtrank((R^R8R_RtR9RAR`RuRaReRvRbRcRwRxRyRds2lib/python2.7/site-packages/sklearn/tree/export.pytexport_graphvizKsrg$o-P               - (t__doc__tnumbersRtnumpyRt externalsRtutils.validationRR&RRRtobjectRtSENTINELR'RR+R(((s2lib/python2.7/site-packages/sklearn/tree/export.pyts   *