riiI'"`dZddlZddlZddlmZddlmZmZmZddl Z ddl m Z ddl mZmZddlmZeZeZedd ddd d d d dddddd deeeefdeededeedeedededede dedeedededed ed!ef d"Zed ddddddd#dd$ deeefde d%eed&eed'eed(eed)eed*eed+ed,ed ed!efd-Zed dddd#dd.deeefde d%eed&eedeed+ed,ed ed!efd/ZdS)0zPlotting Library.N)BytesIO)AnyOptionalUnion)PathLike)Booster_deprecate_positional_args)XGBModelg?zFeature importancezImportance scoreFeaturesweightTz{v}) axheightxlimylimtitlexlabelylabelfmapimportance_typemax_num_featuresgrid show_values values_formatboosterrrrrrrrrrrrrrkwargsreturnc   ddlm}n"#t$r}td|d}~wwxYwt|tr*|| |nTt|tr|| |n't|tr|ntdstdfdD}| t|d | d}nt|d }t|\}}|| d d \}}tjt|}|j||fd |d || durSt||D]B\}}||d zt%|| |d C|||||8t|t,rt|dkrtdndt/|dzf}|||8t|t,rt|dkrtdndt|f}||||||||||||| |S)aPlot importance based on fitted trees. Parameters ---------- booster : Booster or XGBModel instance, or dict taken by Booster.get_fscore() ax : matplotlib Axes Target axes instance. If None, new figure and axes will be created. grid : Turn the axes grids on or off. Default is True (On). importance_type : How the importance is calculated: either "weight", "gain", or "cover" * "weight" is the number of times a feature appears in a tree * "gain" is the average gain of splits which use the feature * "cover" is the average coverage of splits which use the feature where coverage is defined as the number of samples affected by the split max_num_features : Maximum number of top features displayed on plot. If None, all features will be displayed. height : Bar height, passed to ax.barh() xlim : Tuple passed to axes.xlim() ylim : Tuple passed to axes.ylim() title : Axes title. To disable, pass None. xlabel : X axis title label. To disable, pass None. ylabel : Y axis title label. To disable, pass None. fmap : The name of feature map file. show_values : Show values on plot. To disable, pass False. values_format : Format string for values. "v" will be replaced by the value of the feature importance. e.g. Pass "{v:.2f}" in order to limit the number of digits after the decimal point to two, for each value printed on the graph. kwargs : Other keywords passed to ax.barh() Returns ------- ax : matplotlib Axes rNz.You must install matplotlib to plot importance)rrz/tree must be Booster, XGBModel or dict instancez_Booster.get_score() results in empty. This maybe caused by having all trees as decision dumps.c$g|] }||f Sr!).0k importances V/var/www/html/bet.cuttalo.com/ml/venv/lib/python3.11/site-packages/xgboost/plotting.py z#plot_importance..ks" 5 5 5Qq*Q- 5 5 5c|dSNrr!_xs r%z!plot_importance..n r!ur')keyc|dSr)r!r*s r%r,z!plot_importance..pr-r'rcenter)alignrT)v)vaz"xlim must be a tuple of 2 elementsg?z"ylim must be a tuple of 2 elements)matplotlib.pyplotpyplot ImportError isinstancer get_booster get_scorer dict ValueErrorsortedzipsubplotsnparangelenbarhtextfloatformat set_yticksset_yticklabelstuplemaxset_xlimset_ylim set_title set_xlabel set_ylabelr)rrrrrrrrrrrrrrrpltetupleslabelsvalues_ylocsxyr$s @r%plot_importancerZstDS''''''' SSSJKKQRRS'8$$ L((**44+$5   GW % %L&&T&RR GT " "L JKKK   I   6 5 5 5* 5 5 5F#$4$45557G6G6H6HI$4$4555&\NFF z Q""2 Ic&kk " "E BGE6C&CCFCCCd&& M MDAq GGAE588]%9%9A%9%>%>8G L L L LMM%v $&& C#d))q..ABB B+93v;;$%KK $&& C#d))q..ABB B+9CKK KK  U  f  fGGDMMM Is )$)F) r num_treesrankdir yes_colorno_colorcondition_node_paramsleaf_node_params with_statstree_idxr[r\r]r^r_r`rarbc   ddlm} n"#t$r} td| d} ~ wwxYwt|tr|}d} i}| D] \}}|||< |i| d<|| dd<|D]/\}}| dd || d|<ni| d<| |=0||i| d<| || dd <| || dd <||| d <||| d <| r| d z } | tj | z } |1tj dt| d|fvrtd|} ||| || }| |}|S)aConvert specified tree to graphviz instance. IPython can automatically plot the returned graphviz instance. Otherwise, you should call .render() method of the returned graphviz instance. Parameters ---------- booster : Booster or XGBModel instance fmap : The name of feature map file num_trees : .. deprecated:: 3.0 Specify the ordinal number of target tree rankdir : Passed to graphviz via graph_attr yes_color : Edge color when meets the node condition. no_color : Edge color when doesn't meet the node condition. condition_node_params : Condition node configuration for for graphviz. Example: .. code-block:: python {'shape': 'box', 'style': 'filled,rounded', 'fillcolor': '#78bceb'} leaf_node_params : Leaf node configuration for graphviz. Example: .. code-block:: python {'shape': 'box', 'style': 'filled', 'fillcolor': '#e48038'} with_stats : .. versionadded:: 3.0 Controls whether the split statistics should be included. tree_idx : .. versionadded:: 3.0 Specify the ordinal index of target tree. kwargs : Other keywords passed to graphviz graph_attr, e.g. ``graph [ {key} = {value} ]`` Returns ------- graph: graphviz.Source r)Sourcez&You must install graphviz to plot treeNdot graph_attrsr\edger]r^r_r`:zAThe `num_trees` parameter is deprecated, use `tree_idx` insetad. zDBoth `num_trees` and `tree_idx` are used, prefer `tree_idx` instead.)r dump_formatra)graphvizrdr8r9r r:itemsgetjsondumpswarningswarn FutureWarningr=get_dump)rrr[r\r]r^r_r`rarbrrdrR parametersextrar.valuetreegs r% to_graphvizrxs!VK####### KKKBCCJK'8$$(%%''J Ellnn Uc  "}+2}i(kkmm U ::mT * * 6).F= !# & &$&F= ! 3KK 4v&/v{#%-vz"(*?&'#%5!" )c dj(((  O     Ay> ) )V    :*  U U D t A Hs (#()rr[r\rrarbc  ddlm}ddlm} n"#t$r} td| d} ~ wwxYw|| dd\} }t |f|||||d|} t } | | d | d| | }| || d |S) aPlot specified tree. Parameters ---------- booster : Booster or XGBModel instance fmap: str (optional) The name of feature map file num_trees : .. deprecated:: 3.0 rankdir : str, default "TB" Passed to graphviz via graph_attr ax : matplotlib Axes, default None Target axes instance. If None, new figure and axes will be created. with_stats : .. versionadded:: 3.0 See :py:func:`to_graphviz`. tree_idx : .. versionadded:: 3.0 See :py:func:`to_graphviz`. kwargs : Other keywords passed to :py:func:`to_graphviz` Returns ------- ax : matplotlib Axes r)image)r7z(You must install matplotlib to plot treeNr)rr[r\rarbpng)rGoff) matplotlibrzr7r8r@rxrwritepipeseekimreadimshowaxis)rrr[r\rrarbrrzrQrRrVrwsimgs r% plot_treers#bM$$$$$$,,,,,,, MMMDEE1LM z Q""2        A  AGGAFF%F !!!FF1III ,,q//CIIcNNNGGENNN Is  .).)__doc__rmroiortypingrrrnumpyrA_typingrcorer r sklearnr AxesGraphvizSourcer<rFrJstrintboolrZrxrr!r'r%rs ''''''''''55555555   %$#&*AAA 8Wd* +A A  A 5/ A 5/ A A A A AAsmA AAA !A" #AAAAH#!#",0'+@ @ @ 7H$ %@  @ } @ c] @ } @ sm@ $D>@ tn@ @ @ @ @ @ @ @ F#!JJJ 7H$ %J J} J c] J  JJJJ JJJJJJr'