
    Гh                     R    S r SSKrSSKJrJrJr  SSKrSSKrSSK	r
S\S\SS4S jrg)z Tests for training continuation.    N)AnyDictTypeVardevicetree_methodreturnc                   ^ [         R                  " S5      nSnSnUR                  X4SS9u  pV[        R                  " XV5      nUSSSU S	.n[        R
                  " XSS
9n	U	R                  SS9n
[        R
                  " XSS
9n[        R
                  " XSUS9nUR                  SS9n[        S[        [        [        4   [        [        [        [        5      nSUSUSS4U4S jjm[        U
5      [        U5      :X  d   e[        U
5       H<  u  p[         R"                  " U5      n[         R"                  " X   5      nT" UU5        M>     g)zRun training continuation test.zsklearn.datasets@          )random_state2z0.1z0.01)r   	max_depthgammaalphar   )num_boost_roundjson)dump_format)r   	xgb_modelTobj_0obj_1r   Nc                   > [        U [        5      (       a  [        R                  " XSS9(       d   eg [        U [        5      (       a  X:X  d   eg [        U [
        5      (       a  X:X  d   eg [        U [        5      (       a  [        [        U R                  5       5      5       H  n[        U R                  5       5      U   [        UR                  5       5      U   :X  d   e[        U R                  5       5      U   S:w  d  Ma  T" [        U R                  5       5      [        UR                  5       5      5        M     g [        U 5       H  u  p#X   nT" X45        M     g )Ngư>)atolmissing)
isinstancefloatnpisclosestrintdictrangelenitemslistkeysvalues	enumerate)r   r   ilhsrhsrecursive_compares        oC:\Users\julio\OneDrive\Documentos\Trabajo\Ideas Frescas\venv\Lib\site-packages\xgboost/testing/continuation.pyr-   Arun_training_continuation_model_output.<locals>.recursive_compare%   s   eU##::e666s##>!>s##>!>t$$3u{{}-.EJJL)!,UZZ\0B10EEEE

%a(I5%d5<<>&:D<PQ /
 $E*h!#+ +    )pytestimportorskipmake_regressionxgbDMatrixtrainget_dumpr   r   r    r   r   r!   r&   r$   r)   r   loads)r   r   datasets	n_samples
n_featuresXydtrainparamsbst_0dump_0bst_1dump_1r   r*   r+   r   r   r-   s                     @r.   &run_training_continuation_model_outputrD      s3   ""#56HIJ##I#JDA[[F"F IIfb9E^^^/FIIfb9EIIfbEJE^^^/FT#s(^UCd;A, ,1 , ," v;#f+%%%F#

3

69%%' $r0   )__doc__r   typingr   r   r   numpyr   r1   xgboostr4   r    rD    r0   r.   <module>rJ      s4    &  % %   .(3 .(S .(T .(r0   