^jpdZddlZddlmZddlmZddlmZmZddl m Z dd l m Z dd l mZd ed ed dfdZy)z$Test plotting functions for XGBoost.N)Source)Axes) plot_tree to_graphviz) XGBRegressor)make_categorical)Device tree_methoddevicereturnctdddd\}}tdd||}|j|||jj d }|D]?}t j |}d |jvr+t|d tr?Jt|td z }t|tsJt|t|d z } t| tsJy)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_treegraphaxs c/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/xgboost/testing/plotting.pyrun_categoricalr,s D"b 7DAq bkRX CGGAqM OO  & &6 & :EVD!&*Vr7s7*  -""  f  r-