ó
    uñ:iA  ã                   ó¦   • S r SSKJr  SSKrSSKJr  SSKJr  SSK	J
r
  SS	KJr  SS
KJr  SSKJr  S\S\\   S\R"                  4S jrS\SS4S jrg)zTests for inference.é    )ÚTypeN)Úlogité   )ÚDMatrix)Útrainé   )Úvalidate_leaf_output)Úget_basescore)ÚDeviceÚdeviceÚDMatrixTÚreturnc                 ó†  • SnSnSnSnSn[         R                  R                  S5      nUR                  X#5      nUR	                  SXBS9n	U" X‰5      n
[        UUSS	.U
US
9nUR                  SU 05        U" [         R                  " SU4S95      nUR                  USS9nUR                  S   S:X  d   eUR                  U
SSS9nUR                  S   U:X  d   eUR                  S   U:X  d   eUR                  S   U:X  d   eUR                  S   U:X  d   e[        Xå5        [         R                  " S5      nUR                  U
SSU4SS9nUS   n[         R                  " UR                  5      XE-  U-  :X  d   e[        SS0SU
S9nUR                  SU 05        UR                  U
SS9R                  U4:X  d   eU$ )z$Run tests for leaf index prediction.éd   é   é   é
   iÊ  r   )ÚlowÚhighÚsizeÚhist)Únum_parallel_treeÚ	num_classÚtree_method)Únum_boost_roundr   )ÚshapeT)Ú	pred_leaf)r   Ústrict_shaper   r   é   )r   Úiteration_ranger   )r   .r   )r   Údtrain)ÚnpÚrandomÚRandomStateÚrandnÚrandintr   Ú	set_paramÚonesÚpredictr   r	   Úint32Úprod)r   r   ÚrowsÚcolsÚclassesr   r   ÚrngÚXÚyÚmÚboosterÚemptyÚ
empty_leafÚleafÚn_itersÚslicedÚfirsts                     ÚZ/srv/projetos/modelo_ml_acdoc/venv/lib/python3.13/site-packages/xgboost/testing/predict.pyÚrun_predict_leafr;      só  € à€DØ€DØ€GØÐØ€OÜ
�)‰)×
Ñ
 Ó
%€CØ�	‰	�$Ó€AØ�‰˜ ˆÐ3€Aá�‹€AÜà!2Ø Ø!ñ	
ð
 	
Ø'ñ€Gð ×Ñ�x Ð(Ô)Ù”R—W’W A t 9Ñ-Ó.€EØ—‘ °$�Ð7€JØ×Ñ˜AÑ !Ó#Ð#Ð#à�?‰?˜1¨¸4ˆ?Ð@€DØ�:‰:�a‰=˜DÓ Ð Ð Ø�:‰:�a‰=˜OÓ+Ð+Ð+Ø�:‰:�a‰=˜GÓ#Ð#Ð#Ø�:‰:�a‰=Ð-Ó-Ð-Ð-ä˜Ô1ä�hŠh�q‹k€GØ�_‰_Ø	ØØ˜G˜Øð	 ð €Fð �6‰N€Eä�7Š7�5—;‘;Ó 7Ñ#>ÀÑ#HÓHÐHÐHô �] FÐ+¸QÀqÑI€GØ×Ñ�x Ð(Ô)Ø�?‰?˜1¨ˆ?Ð-×3Ñ3¸°wÓ>Ð>Ð>à€Kó    c                 ó,  • SSK Jn  SnU" SS9u  p4[        USU S.[        X45      SS	9n[        R
                  R                  [        U5      U5        UR                  [        X45      5      n[        R                  " UR                  [        U5      [        R                  S
9n[        X4US9n[        SSU S.USS	9n[        R
                  R                  [        U5      S5        UR                  U5      n	[        R
                  R                  Xi5        g)z/Test for the relation between score and margin.r   )Úmake_classificationg      à?ié  )Úrandom_statezbinary:logistic)Ú
base_scoreÚ	objectiver   r   )r!   r   )Ú
fill_valueÚdtype)Úbase_margingš™™™™™É?N)Úsklearn.datasetsr>   r   r   r"   ÚtestingÚassert_allcloser
   r)   Úfullr   r   Úfloat32)
r   r>   Ú	interceptr0   r1   r3   Úpredt_0ÚmarginÚXyÚpredt_1s
             r:   Úrun_base_margin_vs_base_scorerO   F   sã   € å4à€Iá¨DÑ1�D€AÜØ Ð/@ÈFÑSÜ�q‹}Øñ€Gô
 ‡J�J×Ñœ}¨WÓ5°yÔAØ�o‰oœg a›mÓ,€Gä�WŠW�Q—W‘W¬¨yÓ)9ÄÇÁÑL€FÜ	� 6Ñ	*€BäØÐ):ÀfÑMØØñ€Gô
 ‡J�J×Ñœ}¨WÓ5°sÔ;Ø�o‰o˜bÓ!€Gä‡J�J×Ñ˜wÕ0r<   )Ú__doc__Útypingr   Únumpyr"   Úscipy.specialr   Úcorer   Útrainingr   Úsharedr	   Úupdaterr
   Úutilsr   Úndarrayr;   rO   © r<   r:   Ú<module>r[      sV   ðÙ å ã Ý å Ý Ý (Ý "Ý ð3˜Vð 3¨t°G©}ð 3ÀÇÁô 3ðl1¨&ð 1°Tõ 1r<   