^i< dZddlZddlmZddlmZmZmZddlm Z ddl m Z m Z ddl mZmZd d lmZd ed e d e deddf dZdeeeeffdZd eded e d e deddf dZy)zTests for callback functions.N)product)DictListTuple)LearningRateScheduler)BoosterDMatrix)cvtrain)Device tree_methoddtraindtestdevicereturnc t}|df|dfg}d}dddd||d}i}t|||||gd g| } ttt|dd} t | t sJ| d| d kDsJdddd||d }i}t|||||gd g| } ttt|dd} t | t sJ| d| d kDsJddd||d }i}t|||||gdg| } ttt|dd} t | t sJ| d| d k(sJ|fdtdtdtfd} i}t|||||| g| } ttt|dd}t | t sJ|d| dk(sJtdt| D]}||| |k7rJt||||| gy)z=Test learning rate scheduler, used by both CPU and GPU tests.evalr rrbinary:logisticerror) max_deptheta objective eval_metricrr)g?gffffff?g333333?g?)evals callbacks evals_result)r learning_raterrrrrrrrr)rrrrithroundnum_boost_roundrc||dzz SNr )r#r$s Z/var/www/html/hubwallet-dev/venv/lib/python3.12/site-packages/xgboost/testing/callbacks.py eta_decayz run_eta_decay..eta_decay^s(Q,//r )rN) rr listmapfloat isinstancer intrangelenr )rrrr scheduler watchlist num_roundparamrbst eval_errors_0 eval_errors_1 eval_errors_2r) eval_errors_3is r( run_eta_decayr<s&I67"34II &"  E%'L  123!  CUL$8$ABCM c7 ## #  mB/ // /&"  EL  123!  CUL$8$ABCM c7 ## #  mB/ // /&"  EL  \*+!  CUL$8$ABCM c7 ## #  }R0 00 09B0C0#0e0L  Y'(!  CUL$8$ABCM c7 ## #  }Q/ // / 1c-( )4Q=#33334ufiIi,@+ABr*c4ttddggdS)z)Test parameters for the leaf output test.approxhist)rzreg:absoluteerrorreg:quantileerror)r+rr'r*r(tree_methods_objsrAvs$  v     r*rc zdt}|df|dfg}d|d||d}|dk(rd|d <d td tffd }t|||||g } d td tffd } t||||| g } | j d} | j d} t j | }t j | }|ddddd}|ddddd}|d|dk(sJ|d|dk(sJ|ddddd}|ddddd}|d|dk7sJ|d|dk7sJy)z&check decay has effect on leaf output.rrr rrr"r@g333333?quantile_alphar;rc|dzz Sr&r'r;r4s r( eta_decay_0z.run_eta_decay_leaf_output..eta_decay_0sAE""r*)rrc |dkDry|dzz S)Nr g@r'rEs r( eta_decay_1z.run_eta_decay_leaf_output..eta_decay_1s q5AE""r*json) raw_formatlearnergradient_boostermodeltrees base_weightssplit_conditionsN)rr/r-r save_rawrIloads)rrrrrr2r3r5rFbst0rHbst1 bst_json0 bst_json1j0j1 tree_2th_0 tree_2th_1 tree_3th_0 tree_3th_1r4s @r(run_eta_decay_leaf_outputr^sI%I67"34I"  E''"%#s#u#  [)*  D#s#u#  [)*  D  0I  0I I B I BI127;GDQGJI127;GDQGJ n %N)C CC C ( )Z8J-K KK KI127;GDQGJI127;GDQGJ n %N)C CC C ( )Z8J-K KK Kr*)__doc__rI itertoolsrtypingrrrcallbackrcorer r trainingr r utilsrstrr<rAr^r'r*r(rgs# $$,# eCeC%eC.5eC?EeC eCP 4c3h0 8L8L!$8L.58L>E8LOU8L 8Lr*