o ڭj< @sdZddlZddlmZddlmZmZmZddlm Z ddl m Z m Z ddl mZmZd d lmZd ed e d e deddf ddZdeeeeffddZd eded e d e deddf ddZdS)zTests for callback functions.N)product)DictListTuple)LearningRateScheduler)BoosterDMatrix)cvtrain)Device tree_methoddtraindtestdevicereturncCs t}|df|dfg}d}dddd||d}i}t|||||gd g|d } ttt|dd} t| ts8J| d| d ksBJdddd||d }i}t|||||gd g|d } ttt|dd} t| tsnJ| d| d ksxJddd||d }i}t|||||gdg|d } ttt|dd} t| tsJ| d| d ksJ|fdtdtdtfdd} i}t|||||| g|d } ttt|dd}t| tsJ|d| dksJtdt | D] }||| |ksJqt ||||| gddS)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_roundrcSs ||dSNr )r!r"r$r$V/var/www/html/CropPilot/venv/lib/python3.10/site-packages/xgboost/testing/callbacks.py eta_decay^ z run_eta_decay..eta_decayr )rN) rr listmapfloat isinstancerintrangelenr )rrrr scheduler watchlist num_roundparamrbst eval_errors_0 eval_errors_1 eval_errors_2r& eval_errors_3ir$r$r% run_eta_decays   r9cCsttddggdS)z)Test parameters for the leaf output test.approxhist)rzreg:absoluteerrorreg:quantileerror)r(rr$r$r$r%tree_methods_objsvs r=rcstdt}|df|dfg}d|d||d}|dkrd|d <d td tffd d }t|||||gd} d td tffdd } t||||| gd} | jdd} | jdd} t| }t| }|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_alphar8rcs |dSr#r$r8r1r$r% eta_decay_0r'z.run_eta_decay_leaf_output..eta_decay_0)rrcs|dkrdS|dS)Nr g@r$r?r@r$r% eta_decay_1s z.run_eta_decay_leaf_output..eta_decay_1json) raw_formatlearnergradient_boostermodeltrees base_weightssplit_conditionsN)rr,r*r save_rawrCloads)rrrrrr/r0r2rAbst0rBbst1 bst_json0 bst_json1j0j1 tree_2th_0 tree_2th_1 tree_3th_0 tree_3th_1r$r@r%run_eta_decay_leaf_outputsN    rX)__doc__rC itertoolsrtypingrrrcallbackrcorerr trainingr r utilsr strr9r=rXr$r$r$r%sB    h