siirdZddlZddlmZddlmZddlmZmZddl m Z dd l m Z dd l mZd ed ed dfdZdS)z$Test plotting functions for XGBoost.N)Source)Axes) plot_tree to_graphviz) XGBRegressor)make_categorical)Device tree_methoddevicereturncLtdddd\}}tdd||}||||d }|D]I}t j|}d |vst|d tsJJt|t|d z }t|tsJt|t|d z } t| tsJdS)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 ^/var/www/html/bet.cuttalo.com/ml/venv/lib/python3.11/site-packages/xgboost/testing/plotting.pyrun_categoricalr,s' D"b 7 7 7DAq bkRX   CGGAqMMM OO   & &6 & : :EVVD!!&&*Vr7s**  --------""""""""""""  f       r-