o ڭj@srdZddlZddlmZddlmZddlmZmZddl m Z dd l m Z dd l mZd ed ed dfddZdS)z$Test plotting functions for XGBoost.N)Source)Axes) plot_tree to_graphviz) XGBRegressor)make_categorical)Device tree_methoddevicereturnc Cstddddd\}}tdd||d}||||jd d }|D]}t|}d |vs8t|d t s8Jq"t |t |d d}t|t sJJt |t |d d} t| ts[JdS)z2Tests plotting functions for categorical features.iF)onehotT )enable_categorical n_estimatorsr r json) dump_formatleafsplit_conditionr)tree_idxN)r rfit get_boosterget_dumprloadskeys isinstancelistrlenrrr) r r Xyregtreestreej_treegraphaxr)U/var/www/html/CropPilot/venv/lib/python3.10/site-packages/xgboost/testing/plotting.pyrun_categoricals  r+)__doc__rgraphvizrmatplotlib.axesrplottingrrsklearnrdatar utilsr strr+r)r)r)r*s