sii< dZddlZddlmZddlmZmZmZddlm Z ddl m Z m Z ddl mZmZd d lmZd ed e d e deddf dZdeeeeffdZd eded e d e deddf dZdS)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 ksJdddd||d }i}t|||||gd g| } ttt|dd} t | t sJ| d| d ksJddd||d }i}t|||||gdg| } ttt|dd} t | t sJ| d| d ksJ|fdtdtdtfd} i}t|||||| g| } ttt|dd}t | t sJ|d| dksJtdt| D]}||| |ksJt||||| gdS)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 _/var/www/html/bet.cuttalo.com/ml/venv/lib/python3.11/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<se&I67"34II &"   E%'L  9111223!    CUL$8$ABBCCM c7 # ####  mB/ / / / /&"   EL  9111223!    CUL$8$ABBCCM c7 # ####  mB/ / / / /&"   EL  9\\\**+!    CUL$8$ABBCCM c7 # ####  }R0 0 0 0 09B00C0#0e0000L  9Y''(!    CUL$8$ABBCCM c7 # ####  }Q/ / / / / 1c-(( ) )44Q=#333333ufiIIi,@,@+ABBBBBBr*cDttddggdS)z)Test parameters for the leaf output test.approxhist)rzreg:absoluteerrorreg:quantileerror)r+rr'r*r(tree_methods_objsrAvs8  v         r*rc `dt}|df|dfg}d|d||d}|dkrd|d <d td tffd }t|||||g } d td tffd } t||||| g } | d} | d} t j| }t j| }|ddddd}|ddddd}|d|dksJ|d|dksJ|ddddd}|ddddd}|d|dksJ|d|dksJdS)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$|dkrdS|dzz S)Nr g@r'rEs r( eta_decay_1z.run_eta_decay_leaf_output..eta_decay_1s q553AE""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^sHI%I67"34I"   E'''"%#s#u######  9[))*    D#s#u######  9[))*    D  00I  00I I  B I  BI127;GDQGJI127;GDQGJ n %N)C C C C C ( )Z8J-K K K K KI127;GDQGJI127;GDQGJ n %N)C C C C C ( )Z8J-K K K K K K Kr*)__doc__rI itertoolsrtypingrrrcallbackrcorer r trainingr r utilsrstrr<rAr^r'r*r(rgsi## $$$$$$$$$$,,,,,,######## eCeC%eC.5eC?EeC eCeCeCeCP 4c3h0    8L8L!$8L.58L>E8LOU8L 8L8L8L8L8L8Lr*