o ٭j;'"@sdZddlZddlZddlmZddlmZmZmZddl Z ddl m Z ddl mZmZddlmZeZeZedd ddd d d d dddddd deeeefdeededeedeedededede dedeedededed ed!ef d"d#Zed ddddddd$dd% deeefde d&eed'eed(eed)eed*eed+eed,ed-ed ed!efd.d/Zed dddd$dd0deeefde d&eed'eedeed,ed-ed ed!efd1d2ZdS)3zPlotting Library.N)BytesIO)AnyOptionalUnion)PathLike)Booster_deprecate_positional_args)XGBModelg?zFeature importancezImportance scoreFeaturesweightTz{v}) axheightxlimylimtitlexlabelylabelfmapimportance_typemax_num_featuresgrid show_values values_formatboosterrrrrrrrrrrrrrkwargsreturnc  s>zddlm}Wnty}ztd|d}~wwt|tr)|j| |dnt|tr6|j| |dn t|tr>|nt dsHt dfddD}| durct |d d d | d}nt |d d d }t |\}}|dur}| d d \}}t t|}|j||fd|d|| durt ||D]\}}|j|d t|| j|dddq|||||durt|trt|dkrt dndt|df}|||durt|trt|dkrt dndt|f}|||dur|||dur|||dur|||| |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.csg|]}||fqSr).0k importancerM/var/www/html/CropPilot/venv/lib/python3.10/site-packages/xgboost/plotting.py lsz#plot_importance..cS|dSNrr_xrrr#oz!plot_importance..)keycSr%r&rr'rrr#r)qr*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_scorerdict ValueErrorsortedzipsubplotsnparangelenbarhtextfloatformat set_yticksset_yticklabelstuplemaxset_xlimset_ylim set_title set_xlabel set_ylabelr)rrrrrrrrrrrrrrrpltetupleslabelsvalues_ylocsxyrr!r#plot_importancesjB     $            rVF) r num_treesrankdir yes_colorno_colorcondition_node_paramsleaf_node_params with_statstree_idxrWrXrYrZr[r\r]r^c  Kszddlm} Wnty} ztd| d} ~ wwt|tr#|}d} i}| D]\}}|||<q+|durBi| d<|| dd<|D]\}}| dddurY|| d|<ni| d<| |=qF|dusi|durmi| d<|durw|| dd <|dur|| dd <|dur|| d <|dur|| d <| r| d 7} | t | 7} |durt dt | d|fvrt d|} |j|| |d| }| |}|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_attrsrXedgerYrZr[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_formatr])graphvizr_r4r5r r6itemsgetjsondumpswarningswarn FutureWarningr9get_dump)rrrWrXrYrZr[r\r]r^rr_rN parametersextrar+valuetreegrrr# to_graphvizs`K       rs)rrWrXrr]r^c Kszddlm}ddlm} Wnty} ztd| d} ~ ww|dur,| dd\} }t|f|||||d|} t} | | jdd | 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)r3z(You must install matplotlib to plot treeNr)rrWrXr]r^png)rCoff) matplotlibrtr3r4r<rsrwritepipeseekimreadimshowaxis)rrrWrXrr]r^rrtrMrNrRrrsimgrrr# plot_trees61       r)__doc__rhrjiortypingrrrnumpyr=_typingrcorerr sklearnr AxesGraphvizSourcer8rBrFstrintboolrVrsrrrrr#s