^j;'"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/Zy)0zPlotting Library.N)BytesIO)AnyOptionalUnion)PathLike)Booster_deprecate_positional_args)XGBModelg?zFeature importancezImportance scoreFeaturesweightTz{v}) axheightxlimylimtitlexlabelylabelfmapimportance_typemax_num_featuresgrid show_values values_formatboosterrrrrrrrrrrrrrkwargsreturnc   ddlm}t|tr"|j j | |}nBt|tr|j | |}nt|tr|}n td|s td|Dcgc] }|||f }}| t|d| d}nt|d }t|\}}||jd d \}}tjt|}|j ||fd |d || d urEt||D]6\}}|j#|d zt%|| j'|d 8|j)||j+||)t|t,rt|dk7rtddt/|dzf}|j1||)t|t,rt|dk7rtddt|f}|j3|||j5|||j7|||j9||j;| |S#t$r}td|d}~wwxYwcc}w)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 |dSNr_xs [/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/xgboost/plotting.pyz!plot_importance..o r!u)keyc |dSr!r"r#s r%r&z!plot_importance..qr'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)rrrrrrrrrrrrrrrplte importancektupleslabelsvalues_ylocsxys r%plot_importancerWsDS''8$((*44+$5  GW %&&T&R GT " JKK  I  +5 5Qq*Q- 5F 5#$457G6G6HI$45&\NFF z Q"2 IIc&k "E BGGE6C&CFCd& MDAq GGAE58]%9%9A%9%>8G L MMM%v $&#d)q.AB B3v;$%KK $&#d)q.AB BCK KK  U  f  fGGDM I} SJKQRRS(6sI# J# I=, I88I=F) r num_treesrankdir yes_colorno_colorcondition_node_paramsleaf_node_params with_statstree_idxrXrYrZr[r\r]r^r_c   ddlm} t|tr|j }d} i}| j D] \}}|||< | i| d<|| dd<|j D](\}}| jdd || d|<ni| d<| |=*||i| d<||| dd <||| dd <||| d <||| d <| r| d z } | tj| z } |-tjdt| d|fvr td|} |j|| || }| |}|S#t$r} td| d} ~ wwxYw)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_attrsrYedgerZr[r\r]:zAThe `num_trees` parameter is deprecated, use `tree_idx` instead. zDBoth `num_trees` and `tree_idx` are used, prefer `tree_idx` instead.)r dump_formatr^)graphvizrar3r4r r5itemsgetjsondumpswarningswarn FutureWarningr8get_dump)rrrXrYrZr[r\r]r^r_rrarM parametersextrar)valuetreegs r% to_graphvizrusVK#'8$%%'J Elln Uc  "}+2}i(kkm U ::mT * 6).F= !# &$&F= ! 3K  4v&/v{#%-vz"(*?&'#%5!" c djj((  O   Ay> )V    :*  U D t A Hi KBCJKsD-- E6 EE)rrXrYrr^r_c  ddlm}ddlm} || j dd\} }t |f|||||d|} t } | j| jd | jd|j| }|j||jd |S#t$r} td| d} ~ wwxYw) 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)r2z(You must install matplotlib to plot treeNr)rrXrYr^r_png)rBoff) matplotlibrwr2r3r;rurwritepipeseekimreadimshowaxis)rrrXrYrr^r_rrwrLrMrSrtsimgs r% plot_treersbM$, z Q"2      A  AGGAFF%F !FF1I ,,q/CIIcNGGEN I/ MDE1LMs B(( C1 B==C)__doc__rjrliortypingrrrnumpyr<_typingrcorer r sklearnr AxesGraphvizSourcer7rArEstrintboolrWrurr"r(r%rs ''5   %$#&*A 8Wd* +A A  A 5/ A 5/ A A A A AAsmA AAA !A" #AAH#!#",0'+@ 7H$ %@  @ } @ c] @ } @ sm@ $D>@ tn@ @ @ @ @ @ F#!J 7H$ %J J} J c] J  JJJJ JJr(