
    sii                     R    d Z ddlZddlmZmZmZ ddlZddlZddl	Z
dededdfdZdS )z Tests for training continuation.    N)AnyDictTypeVardevicetree_methodreturnc                    t          j        d          }d}d}|                    ||d          \  }}t          j        ||          }|ddd| d	}t          j        ||d
          }	|	                    d          }
t          j        ||d
          }t          j        ||d|          }|                    d          }t          dt          t          t          f         t          t          t          t                    }d|d|ddffdt          |
          t          |          k    sJ t          |
          D ]?\  }}t!          j        |          }t!          j        ||                   } ||           @dS )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                 T   t          | t                    rt          j        | |d          sJ d S t          | t                    r
| |k    sJ d S t          | t
                    r
| |k    sJ d S t          | t                    rt          t          | 	                                                    D ]}t          |                                           |         t          |                                          |         k    sJ t          |                                           |         dk    rJ t          |                                           t          |                                                     d S t          |           D ]\  }}||         } ||           d S )Ngư>)atolmissing)
isinstancefloatnpisclosestrintdictrangelenitemslistkeysvalues	enumerate)r   r   ilhsrhsrecursive_compares        b/var/www/html/bet.cuttalo.com/ml/venv/lib/python3.11/site-packages/xgboost/testing/continuation.pyr-   zArun_training_continuation_model_output.<locals>.recursive_compare%   s   eU## 	,:eU66666666s## 	,E>>>>>>s## 
	,E>>>>>>t$$ 	,3u{{}}--.. R REJJLL))!,UZZ\\0B0B10EEEEE

%%a(I55%%d5<<>>&:&:D<P<PQQQR R
 $E** , ,3Ah!!#s++++, ,    )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_outputrC      s   "#566HIJ##Iz#JJDAq[AF" F Iffb999E^^^//FIffb999EIffbEJJJE^^^//FT#s(^UCd;;A, ,1 , , , , , , ," v;;#f++%%%%F## ( (3
3
6!9%%%''''( (r/   )__doc__r   typingr   r   r   numpyr   r0   xgboostr3   r    rC    r/   r.   <module>rI      s    & &  % % % % % % % % % %         .(3 .(S .(T .( .( .( .( .( .(r/   