§
    rŠtjÎy  ã                   ó„  — d dl Zd dlZd dlmZmZ d dlmZ d dlm	Z	 d dl
mZmZmZ d dlmZ d dlmZmZmZmZ d dlmZ d d	lmZmZ d d
lmZmZmZmZmZ d dl m!Z!m"Z"m#Z#m$Z$m%Z% dZ&d„ Z'ej(         )                    dej*        ej+        g¦  «        d„ ¦   «         Z,ej(         )                    ddge#z   ¦  «        ej(         )                    dddg¦  «        d„ ¦   «         ¦   «         Z-d„ Z.ej(         )                    dddg¦  «        d„ ¦   «         Z/d„ Z0d„ Z1ej(         )                    dddg¦  «        ej(         )                    de#¦  «        d„ ¦   «         ¦   «         Z2d„ Z3ej(         )                    de!¦  «        d „ ¦   «         Z4d!„ Z5d"„ Z6d#„ Z7d$„ Z8ej(         )                    ddge#z   ¦  «        ej(         )                    d%ddg¦  «        d&„ ¦   «         ¦   «         Z9d'„ Z:ej(         )                    ddge"z   ¦  «        d(„ ¦   «         Z;ej(         )                    d)ddg¦  «        ej(         )                    ddge#z   ¦  «        d*„ ¦   «         ¦   «         Z<ej(         )                    d+e$¦  «        d,„ ¦   «         Z=ej(         )                    de#¦  «        d-„ ¦   «         Z>ej(         )                    ddge#z   ¦  «        ej(         )                    d.d/¦  «        ej(         )                    d0d1¦  «        d2„ ¦   «         ¦   «         ¦   «         Z?ej(         )                    d)ddg¦  «        ej(         )                    dddg¦  «        d3„ ¦   «         ¦   «         Z@d4„ ZAej(         )                    d5dd6g¦  «        ej(         )                    ddge#z   ¦  «        d7„ ¦   «         ¦   «         ZBej(         )                    de#¦  «        d8„ ¦   «         ZCej(         )                    d9g d:¢¦  «        ej(         )                    ddgd;„ e#D ¦   «         z   ¦  «        ej(         )                    dddg¦  «        d<„ ¦   «         ¦   «         ¦   «         ZDed=„ ¦   «         ZEd>„ ZFdS )?é    N)ÚlinalgÚsparse)Úconfig_context)ÚBaseEstimator)Ú	load_irisÚmake_regressionÚmake_sparse_uncorrelated)ÚLinearRegression)ÚLinearClassifierMixinÚ_preprocess_dataÚ_rescale_dataÚmake_dataset)Úadd_dummy_feature)Úget_namespace_and_deviceÚmove_estimator_to)Ú_array_api_for_testsÚassert_allcloseÚassert_array_almost_equalÚassert_array_equalÚ'skip_if_array_api_compat_not_configured)ÚCOO_CONTAINERSÚCSC_CONTAINERSÚCSR_CONTAINERSÚLIL_CONTAINERSÚ_sparse_eye_arrayç�íµ ÷Æ°>c                  óú  — dgdgg} ddg}t          ¦   «         }|                     | |¦  «         t          |j        dg¦  «         t          |j        dg¦  «         t          |                     | ¦  «        ddg¦  «         dgg} dg}t          ¦   «         }|                     | |¦  «         t          |j        dg¦  «         t          |j        dg¦  «         t          |                     | ¦  «        dg¦  «         d S )Né   é   r   )r
   Úfitr   Úcoef_Ú
intercept_Úpredict)ÚXÚYÚregs      úb/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/sklearn/linear_model/tests/test_base.pyÚtest_linear_regressionr(   &   sô   € ð ˆˆqˆcˆ
€AØ	
ˆAˆ€Aå
Ñ
Ô
€CØ‡G‚GˆAˆq�M„M€Må˜cœi¨!¨Ñ-Ô-Ð-Ý˜cœn¨q¨cÑ2Ô2Ð2Ý˜cŸkšk¨!™nœn¨q°!¨fÑ5Ô5Ð5ð ˆˆ€AØ	
ˆ€Aå
Ñ
Ô
€CØ‡G‚GˆAˆq�M„M€MÝ˜cœi¨!¨Ñ-Ô-Ð-Ý˜cœn¨q¨cÑ2Ô2Ð2Ý˜cŸkšk¨!™nœn¨q¨cÑ2Ô2Ð2Ð2Ð2ó    Údtypec                 óÂ  — t           j                             d¦  «        }d}|                     |¦  «        }d|z  d|                     |¦  «        z  z   }t          j        ||g¦  «        }|ddgz  d|                     |¦  «        z  z   }|                     | ¦  «        }|                     | ¦  «        }t          j        ||¦  «        d         }t          d¬	¦  «         	                    ||¦  «        j
        }t           j                             |||z  z
  ¦  «        }	t           j                             |||z  z
  ¦  «        }
|
t          j        |	d
¬¦  «        k    sJ ‚dS )zs
    Check that LinearRegression is as good as `scipy.linalg.lstsq`.
    Non regression test for issue #33032.
    iq  i ¡ g333333Ó?çš™™™™™¹?ç      à?ç       @r   F©Úfit_interceptr   )ÚrelN)ÚnpÚrandomÚRandomStateÚrandÚcolumn_stackÚastyper   Úlstsqr
   r    r!   ÚnormÚpytestÚapprox)r*   ÚrngÚ	n_samplesÚx1Úx2r$   ÚyÚ
coef_scipyÚcoef_sklearnÚ
rmse_scipyÚrmse_sklearns              r'   Útest_linear_regression_vs_lstsqrE   >   s:  € õ Œ)×
Ò
 Ñ
%Ô
%€CØ€Ià	�Š�)Ñ	Ô	€BØ	ˆr‰�C˜#Ÿ(š( 9Ñ-Ô-Ñ-Ñ	-€BÝ
Œ˜˜R˜Ñ!Ô!€AØ	ˆS�#ˆJ‰˜˜sŸxšx¨	Ñ2Ô2Ñ2Ñ2€Aà	�Š�‰Œ€AØ	�Š�‰Œ€Aå”˜a Ñ#Ô# AÔ&€JÝ#°%Ð8Ñ8Ô8×<Ò<¸QÀÑBÔBÔH€Lå”—’  A¨
¡NÑ 2Ñ3Ô3€JÝ”9—>’> ! a¨,Ñ&6Ñ"6Ñ7Ô7€LØ�6œ=¨¸Ð>Ñ>Ô>Ò>Ð>Ð>Ð>Ð>Ð>r)   Úsparse_containerr0   TFc                 óÐ  — t           j                             |¦  «        }d\  }}|                     ||f¬¦  «        }| � | |¦  «        }|                     |¬¦  «        }d|                     |¬¦  «        z   }t          |d¬¦  «        }	|	                     |||¬¦  «         |	j        }
|	j        }|	j        j	        |j	        d         fk    sJ ‚t          j
        |¦  «        }|s|nt          |¦  «        }|j        |z  |z  }|j        |z  |z  }t          j        ||¦  «        }|st          |
|¦  «         d S t          |
|dd …         ¦  «         t          ||d         ¦  «         d S )	N)é   é   ©Úsizeç      ð?g¼‰Ø—²Òœ<)r0   Útol©Úsample_weightr   r   )r2   r3   r4   ÚnormalÚuniformr
   r    r!   r"   ÚshapeÚdiagr   ÚTr   Úsolver   )rF   r0   Úglobal_random_seedr<   r=   Ú
n_featuresr$   r@   rO   r&   Úcoefs1Úinter1ÚWÚX_augÚXwÚywÚcoefs2s                    r'   Ú%test_linear_regression_sample_weightsr_   W   s~  € õ
 Œ)×
Ò
Ð 2Ñ
3Ô
3€Cð !Ñ€Iˆzà�
Š
˜ JÐ/ˆ
Ñ0Ô0€AØÐ#ØÐ˜QÑÔˆØ�
Š
˜	ˆ
Ñ"Ô"€Aà˜#Ÿ+š+¨9˜+Ñ5Ô5Ñ5€Mõ ¨¸EÐ
BÑ
BÔ
B€CØ‡G‚GˆAˆq €GÑ.Ô.Ð.ØŒY€FØŒ^€FàŒ9Œ?˜qœw qœz˜mÒ+Ð+Ð+Ð+õ 	Œ�ÑÔ€AØ"Ð<ˆAˆAÕ(9¸!Ñ(<Ô(<€Eà	Œ�1‰�uÑ	€BØ	Œ�1‰�q‰€BÝŒ\˜"˜bÑ!Ô!€Fàð +Ý˜ Ñ'Ô'Ð'Ð'Ð'å˜  q r r¤
Ñ+Ô+Ð+Ý˜  q¤	Ñ*Ô*Ð*Ð*Ð*r)   c                  ó  — d} t          d¦  «        }t          j        d¦  «        }t          d¬¦  «        }t	          j        t          | ¬¦  «        5  |                     ||¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )Nz9Sparse data was passed for X, but dense data is required.é
   T©Úpositive©Úmatch)r   r2   Úonesr
   r:   ÚraisesÚ	TypeErrorr    )Ú	error_msgr$   r@   r&   s       r'   Ú.test_raises_value_error_if_positive_and_sparserj   €   s»   € ØK€Iå˜"ÑÔ€AÝ
Œ�‰Œ€Aå
 DÐ
)Ñ
)Ô
)€Cå	Œ•y¨	Ð	2Ñ	2Ô	2ð ð Ø�Š��1‰Œˆðð ð ñ ô ð ð ð ð ð ð ð øøøð ð ð ð ð ð s   ÁA5Á5A9Á<A9zn_samples, n_features)r   é   )rk   r   c                 ó~  — t           j                             d¦  «        }|                     | |¦  «        }|                     | ¦  «        }|                     | ¦  «        dz  dz   }d}d}t	          ¦   «         }|                     |||¦  «         |                     |||¦  «         |                     |||¦  «         d S )Nr   r   r   rL   r.   )r2   r3   r4   Úrandnr
   r    )	r=   rW   r<   r$   r@   Úsample_weights_OKÚsample_weights_OK_1Úsample_weights_OK_2r&   s	            r'   Ú9test_raises_value_error_if_sample_weights_greater_than_1drq   Œ   s½   € õ Œ)×
Ò
 Ñ
"Ô
"€CØ�	Š	�)˜ZÑ(Ô(€AØ�	Š	�)ÑÔ€AØŸ	š	 )Ñ,Ô,°Ñ1°AÑ5ÐØÐØÐå
Ñ
Ô
€Cð ‡G‚GˆAˆqÐ#Ñ$Ô$Ð$Ø‡G‚GˆAˆqÐ%Ñ&Ô&Ð&Ø‡G‚GˆAˆqÐ%Ñ&Ô&Ð&Ð&Ð&r)   c                  óZ  — t          j        ddgddgg¦  «        } t          j        g d¢g d¢g¦  «        }t          j        ddg¦  «        }t          d¬	¦  «                             | |¦  «        }t          ¦   «                              | |¦  «        }t          d¬	¦  «                             ||¦  «        }t          ¦   «                              ||¦  «        }|j        j        |j        j        k    sJ ‚|j        j        |j        j        k    sJ ‚|j        j        |j        j        k    sJ ‚d S )
NgmâG©B‹Ø?gÊ\«^ºã?gñ‰L+KÕâ?gìf©iUÚ?)g&Ï/)Â¶Ñ?g ö<E/Ÿæ?g²MdÅô­�?)g".ÆáHwµ?g€%ü÷…|Ê?gÜ0Èeõ±æ?r   Fr/   )r2   Úarrayr
   r    r!   rR   Úndim)ÚX2ÚX3r@   Úlr2_without_interceptÚlr2_with_interceptÚlr3_without_interceptÚlr3_with_intercepts          r'   Útest_fit_interceptr{   ž   s5  € å	Œ�J 
Ð+¨j¸*Ð-EÐFÑ	GÔ	G€BÝ	ŒØ	-Ð	-Ð	-Ð/SÐ/SÐ/SÐTñ
ô 
€Bõ 	Œ�!�Q�ÑÔ€Aå,¸5ÐAÑAÔA×EÒEÀbÈ!ÑLÔLÐÝ)Ñ+Ô+×/Ò/°°AÑ6Ô6Ðå,¸5ÐAÑAÔA×EÒEÀbÈ!ÑLÔLÐÝ)Ñ+Ô+×/Ò/°°AÑ6Ô6ÐàÔ#Ô)Ð-BÔ-HÔ-NÒNÐNÐNÐNØÔ#Ô)Ð-BÔ-HÔ-NÒNÐNÐNÐNØ Ô&Ô+Ð/DÔ/JÔ/OÒOÐOÐOÐOÐOÐOr)   c                 ó²  — t           j                             | ¦  «        }d}t          ||¦  «        }|                     |¦  «        }||z  }t          ¦   «         }|                     ||                     ¦   «         ¦  «         t          ||j	        |j
        z   ¦  «         t          |                     |¦  «        |                     ¦   «         z
  d¦  «         d S )Néd   r   )r2   r3   r4   r   r5   r
   r    Úravelr   r!   r"   r#   )rV   r<   Únr$   Úbetar@   Úolss          r'   Útest_linear_regression_sparser‚   ±   s±   € å
Œ)×
Ò
Ð 2Ñ
3Ô
3€CØ€AÝ˜!˜QÑÔ€AØ�8Š8�A‰;Œ;€DØ	ˆD‰€Aå
Ñ
Ô
€CØ‡G‚GˆAˆq�wŠw‰yŒyÑÔÐÝ˜d C¤I°´Ñ$>Ñ?Ô?Ð?å˜cŸkšk¨!™nœn¨q¯wªw©y¬yÑ8¸!Ñ<Ô<Ð<Ð<Ð<r)   Úcsr_containerc                 óî  — t           j                             d¦  «        }d}d}|                     ||¦  «        }d||dk     <    ||¦  «        }|                     |¦  «        }t          | ¬¦  «        }t          di |¤Ž}	t          di |¤Ž}
|	                     ||¦  «         |
                     ||¦  «         |	j        t          j
        |
j        ¦  «        k    sJ ‚t          |	j        |
j        ¦  «         d S )Nr   éÈ   r   ç        r,   r/   © )r2   r3   r4   rm   r5   Údictr
   r    r"   r:   r;   r   r!   )r0   rƒ   r<   r=   rW   r$   ÚXcsrr@   ÚparamsÚ	clf_denseÚ
clf_sparses              r'   Ú)test_linear_regression_sparse_equal_denser�   À   sý   € õ Œ)×
Ò
 Ñ
"Ô
"€CØ€IØ€JØ�	Š	�)˜ZÑ(Ô(€AØ€A€aˆ#‚g�JØˆ=˜ÑÔ€DØ�Š�ÑÔ€AÝ Ð.Ñ.Ô.€FÝ Ð*Ð* 6Ð*Ð*€IÝ!Ð+Ð+ FÐ+Ð+€JØ‡M‚M�!�QÑÔÐØ‡N‚N�4˜ÑÔÐØÔ¥6¤=°Ô1FÑ#GÔ#GÒGÐGÐGÐGÝ�I”O ZÔ%5Ñ6Ô6Ð6Ð6Ð6r)   c                  ó   — t           j                             d¦  «        } t          | ¬¦  «        \  }}t          j        ||f¦  «        j        }|j        d         }t          ¦   «         }|                     ||¦  «         |j	        j        d|fk    sJ ‚| 
                    |¦  «        }|                     ||¦  «         | 
                    |¦  «        }t          t          j        ||f¦  «        j        |d¬¦  «         d S )Nr   ©Úrandom_stater   r   rk   ©Údecimal)r2   r3   r4   r   ÚvstackrT   rR   r
   r    r!   r#   r   )r<   r$   r@   r%   rW   r&   ÚY_predÚy_preds           r'   Ú'test_linear_regression_multiple_outcomer–   Ô   sÝ   € å
Œ)×
Ò
 Ñ
"Ô
"€CÝ¨Ð,Ñ,Ô,�D€A€qå
Œ	�1�a�&ÑÔÔ€AØ”˜”€Jå
Ñ
Ô
€CØ‡G‚GˆQ��O„O€OØŒ9Œ?˜q *˜oÒ-Ð-Ð-Ð-Ø�[Š[˜‰^Œ^€FØ‡G‚GˆAˆq�M„M€MØ�[Š[˜‰^Œ^€FÝ�bœi¨°Ð(8Ñ9Ô9Ô;¸VÈQÐOÑOÔOÐOÐOÐOr)   Úcoo_containerc                 ó:  — t           j                             | ¦  «        }t          |¬¦  «        \  }} ||¦  «        }t          j        ||f¦  «        j        }|j        d         }t          ¦   «         }|                     ||¦  «         |j	        j        d|fk    sJ ‚| 
                    |¦  «        }|                     ||                     ¦   «         ¦  «         | 
                    |¦  «        }	t          t          j        |	|	f¦  «        j        |d¬¦  «         d S )Nr�   r   r   rk   r‘   )r2   r3   r4   r	   r“   rT   rR   r
   r    r!   r#   r~   r   )
rV   r—   r<   r$   r@   r%   rW   r�   r”   r•   s
             r'   Ú.test_linear_regression_sparse_multiple_outcomer™   å   sú   € õ Œ)×
Ò
Ð 2Ñ
3Ô
3€CÝ#°Ð5Ñ5Ô5�D€A€qØˆ�aÑÔ€AÝ
Œ	�1�a�&ÑÔÔ€AØ”˜”€Jå
Ñ
Ô
€CØ‡G‚GˆAˆq�M„M€MØŒ9Œ?˜q *˜oÒ-Ð-Ð-Ð-Ø�[Š[˜‰^Œ^€FØ‡G‚GˆAˆq�wŠw‰yŒyÑÔÐØ�[Š[˜‰^Œ^€FÝ�bœi¨°Ð(8Ñ9Ô9Ô;¸VÈQÐOÑOÔOÐOÐOÐOr)   c                  ó  — dgdgg} ddg}t          d¬¦  «        }|                     | |¦  «         t          |j        dg¦  «         t          |j        dg¦  «         t          |                     | ¦  «        ddg¦  «         dgg} dg}t          d¬¦  «        }|                     | |¦  «         t          |j        dg¦  «         t          |j        dg¦  «         t          |                     | ¦  «        dg¦  «         d S )Nr   r   Trb   r   )r
   r    r   r!   r"   r#   r   )r$   r@   r&   s      r'   Útest_linear_regression_positiver›   ÷   sü   € à
ˆˆqˆcˆ
€AØ	
ˆAˆ€Aå
 DÐ
)Ñ
)Ô
)€CØ‡G‚GˆAˆq�M„M€Må˜cœi¨!¨Ñ-Ô-Ð-Ý˜cœn¨q¨cÑ2Ô2Ð2Ý˜cŸkšk¨!™nœn¨q°!¨fÑ5Ô5Ð5ð ˆˆ€AØ	
ˆ€Aå
 DÐ
)Ñ
)Ô
)€CØ‡G‚GˆAˆq�M„M€MÝ�C”I ˜sÑ#Ô#Ð#Ý�C”N Q CÑ(Ô(Ð(Ý�C—K’K ‘N”N Q CÑ(Ô(Ð(Ð(Ð(r)   c                 ób  — t           j                             | ¦  «        }t          |¬¦  «        \  }}t          j        ||f¦  «        j        }|j        d         }t          d¬¦  «        }|                     ||¦  «         |j	        j        d|fk    sJ ‚t          j
        |j	        dk    ¦  «        sJ ‚|                     |¦  «        }|                     ||                     ¦   «         ¦  «         |                     |¦  «        }t          t          j        ||f¦  «        j        |¦  «         d S )Nr�   r   Trb   r   r†   )r2   r3   r4   r	   r“   rT   rR   r
   r    r!   Úallr#   r~   r   )	rV   r<   r$   r@   r%   rW   r�   r”   r•   s	            r'   Ú0test_linear_regression_positive_multiple_outcomerž     s  € å
Œ)×
Ò
Ð 2Ñ
3Ô
3€CÝ#°Ð5Ñ5Ô5�D€A€qÝ
Œ	�1�a�&ÑÔÔ€AØ”˜”€Jå
 DÐ
)Ñ
)Ô
)€CØ‡G‚GˆAˆq�M„M€MØŒ9Œ?˜q *˜oÒ-Ð-Ð-Ð-ÝŒ6�#”)˜sÒ"Ñ#Ô#Ð#Ð#Ð#Ø�[Š[˜‰^Œ^€FØ‡G‚GˆAˆq�wŠw‰yŒyÑÔÐØ�[Š[˜‰^Œ^€FÝ•B”I˜v vÐ.Ñ/Ô/Ô1°6Ñ:Ô:Ð:Ð:Ð:r)   c                 óV  — t           j                             | ¦  «        }t          |¬¦  «        \  }}t	          d¬¦  «        }|                     ||¦  «         t	          d¬¦  «        }|                     ||¦  «         t          j        |j        |j        z
  dz  ¦  «        dk    sJ ‚d S )Nr�   Trb   Fr   çü©ñÒMbP?)r2   r3   r4   r	   r
   r    Úmeanr!   )rV   r<   r$   r@   r&   Úregns         r'   Ú.test_linear_regression_positive_vs_nonpositiver£     s�   € å
Œ)×
Ò
Ð 2Ñ
3Ô
3€CÝ#°Ð5Ñ5Ô5�D€A€qå
 DÐ
)Ñ
)Ô
)€CØ‡G‚GˆAˆq�M„M€MÝ UÐ+Ñ+Ô+€DØ‡H‚HˆQ��N„N€NåŒ7�C”I ¤
Ñ*¨qÑ0Ñ1Ô1°DÒ8Ð8Ð8Ð8Ð8Ð8r)   c                 óÜ  — t           j                             | ¦  «        }d}d}|                     ||¦  «        }|d d …df         d|d d …df         z  z   d|d d …df         z  z   d|d d …df         z  z   }t	          d¬	¦  «        }|                     ||¦  «         t	          d
¬	¦  «        }|                     ||¦  «         t          j        |j        |j        z
  dz  ¦  «        dk     sJ ‚d S )Nr…   é   r   r   r   rk   g      ø?Trb   Fr   )r2   r3   r4   r5   r
   r    r¡   r!   )rV   r<   r=   rW   r$   r@   r&   r¢   s           r'   Ú<test_linear_regression_positive_vs_nonpositive_when_positiver¦   ,  sø   € õ Œ)×
Ò
Ð 2Ñ
3Ô
3€CØ€IØ€JØ�Š�˜JÑ'Ô'€AØ	ˆ!ˆ!ˆ!ˆQˆ$Œ�!�a˜˜˜˜1˜”g‘+Ñ  A a a a¨ d¤G¡Ñ+¨c°A°a°a°a¸°d´G©mÑ;€Aå
 DÐ
)Ñ
)Ô
)€CØ‡G‚GˆAˆq�M„M€MÝ UÐ+Ñ+Ô+€DØ‡H‚HˆQ��N„N€NåŒ7�C”I ¤
Ñ*¨qÑ0Ñ1Ô1°DÒ8Ð8Ð8Ð8Ð8Ð8r)   Úuse_swc                 ó�  — t           j                             |¦  «        }|                     dd¦  «        }|                     dd¦  «        }|                     d¦  «        }| � | |¦  «        }n|                     ¦   «         }|                     ¦   «         }|r|                     ¦   «         }	nd }	t          ¦   «         }
|
                     |||	¬¦  «         | �#t          | 	                    ¦   «         |¦  «         nt          ||¦  «         t          ||¦  «         |rt          |	|¦  «         t          d¬¦  «        }
|
                     |||	¬¦  «         | �#t          | 	                    ¦   «         |¦  «         n(t           j
                             ||z
  ¦  «        dk    sJ ‚t          ||¦  «         |rt          |	|¦  «         d S d S )Nra   é   r   rN   F)Úcopy_Xgáz®GáÚ?)r2   r3   r4   rm   r5   Úcopyr
   r    r   Útoarrayr   r9   )rF   r§   rV   r<   Úoriginal_X_dataÚoriginal_y_dataÚoriginal_sw_datar$   r@   rO   r&   s              r'   Útest_inplace_data_preprocessingr°   =  sÙ  € õ
 Œ)×
Ò
Ð 2Ñ
3Ô
3€CØ—i’i  BÑ'Ô'€OØ—i’i  AÑ&Ô&€OØ—x’x ‘|”|ÐàÐ#ØÐ˜_Ñ-Ô-ˆˆà× Ò Ñ"Ô"ˆØ×ÒÑÔ€Að ð Ø(×-Ò-Ñ/Ô/ˆˆàˆõ Ñ
Ô
€CØ‡G‚GˆAˆq €GÑ.Ô.Ð.ØÐ#Ý˜Ÿ	š	™œ _Ñ5Ô5Ð5Ð5å˜˜?Ñ+Ô+Ð+Ý�A�Ñ'Ô'Ð'àð 9Ý˜Ð'7Ñ8Ô8Ð8õ  %Ð
(Ñ
(Ô
(€CØ‡G‚GˆAˆq €GÑ.Ô.Ð.ØÐ#õ 	˜Ÿ	š	™œ _Ñ5Ô5Ð5Ð5õ
 Œy�~Š~˜a /Ñ1Ñ2Ô2°TÒ9Ð9Ð9Ð9õ �A�Ñ'Ô'Ð'àð 9å˜Ð'7Ñ8Ô8Ð8Ð8Ð8ð9ð 9r)   c                 ó&  — t           j                             | ¦  «        }d}d}|                     ||¦  «        }|                     |¦  «        }t          j        |d¬¦  «        }t          j        |d¬¦  «        }t          ||d¬¦  «        \  }}	}
}}}t          |
t          j        |¦  «        ¦  «         t          |d¦  «         t          |t          j        |¦  «        ¦  «         |�J ‚t          ||¦  «         t          |	|¦  «         t          ||d¬¦  «        \  }}	}
}}}t          |
|¦  «         t          ||¦  «         t          |t          j        |¦  «        ¦  «         |�J ‚t          |||z
  ¦  «         t          |	||z
  ¦  «         d S )Nr…   r   r   ©ÚaxisFr/   T)	r2   r3   r4   r5   r¡   r   r   Úzerosrf   )rV   r<   r=   rW   r$   r@   Úexpected_X_meanÚexpected_y_meanÚXtÚytÚX_meanÚy_meanÚX_scaleÚsqrt_sws                 r'   Útest_preprocess_datar½   u  s¢  € Ý
Œ)×
Ò
Ð 2Ñ
3Ô
3€CØ€IØ€JØ�Š�˜JÑ'Ô'€AØ�Š�ÑÔ€AÝ”g˜a aÐ(Ñ(Ô(€OÝ”g˜a aÐ(Ñ(Ô(€Oå/?Ø	ˆ1˜Eð0ñ 0ô 0Ñ,€BˆˆF�F˜G Wõ ˜f¥b¤h¨zÑ&:Ô&:Ñ;Ô;Ð;Ý˜f aÑ(Ô(Ð(Ý˜g¥r¤w¨zÑ':Ô':Ñ;Ô;Ð;Øˆ?ˆ?ˆ?Ý˜b !Ñ$Ô$Ð$Ý˜b !Ñ$Ô$Ð$å/?Ø	ˆ1˜Dð0ñ 0ô 0Ñ,€BˆˆF�F˜G Wõ ˜f oÑ6Ô6Ð6Ý˜f oÑ6Ô6Ð6Ý˜g¥r¤w¨zÑ':Ô':Ñ;Ô;Ð;Øˆ?ˆ?ˆ?Ý˜b ! oÑ"5Ñ6Ô6Ð6Ý˜b ! oÑ"5Ñ6Ô6Ð6Ð6Ð6r)   c                 óü  — t           j                             | ¦  «        }d}d}d}|                     ||¦  «        }|                     ||¦  «        }t          j        |d¬¦  «        }|� ||¦  «        }t          ||d¬¦  «        \  }	}
}	}}	}	t          |t          j        |¦  «        ¦  «         t          |
|¦  «         t          ||d¬¦  «        \  }	}
}	}}	}	t          ||¦  «         t          |
||z
  ¦  «         d S )	Nr…   rk   r   r   r²   Fr/   T)r2   r3   r4   r5   r¡   r   r   r´   )rV   rF   r<   r=   rW   Ú	n_outputsr$   r@   r¶   Ú_r¸   rº   s               r'   Ú test_preprocess_data_multioutputrÁ   “  s  € å
Œ)×
Ò
Ð 2Ñ
3Ô
3€CØ€IØ€JØ€IØ�Š�˜JÑ'Ô'€AØ�Š�˜IÑ&Ô&€AÝ”g˜a aÐ(Ñ(Ô(€OàÐ#ØÐ˜QÑÔˆå-¨a°À%ÐHÑHÔHÑ€A€rˆ1ˆf�a˜Ý˜f¥b¤h¨yÑ&9Ô&9Ñ:Ô:Ð:Ý˜b !Ñ$Ô$Ð$å-¨a°À$ÐGÑGÔGÑ€A€rˆ1ˆf�a˜Ý˜f oÑ6Ô6Ð6Ý˜b ! f¡*Ñ-Ô-Ð-Ð-Ð-r)   Úrescale_with_swc                 ó   — t           j                             |¦  «        }d}d}|                     ||¦  «        }d||dk     <   |d d …dfxx         dz  cc<   d|d d …df<   d|d d …d	f<   |                     |¦  «        }t          j        |                     |¦  «        ¦  «        d
z   }t          j        |d|¬¦  «        }	t          j        |d|¬¦  «        }
t          j        ||d¬¦  «        }t          j        ||z
  dz  |d¬¦  «        }|dt          j        |j        ¦  «        j        z  k     }t          |g d¢¦  «         t          j
        |¦  «        t          j
        |                     ¦   «         ¦  «        z  }d
||<   |� ||¦  «        }t          ||d|| ¬¦  «        \  }}}}}}|�(|                     ¦   «         }|                     ¦   «         }t          ||	¦  «         t          ||
¦  «         t          |t          j        |¦  «        ¦  «         | rzt!          |t          j
        |¦  «        ¦  «         |�t!          ||d d …d f         |z  ¦  «         n t!          ||d d …d f         ||	z
  z  ¦  «         t!          ||||
z
  z  ¦  «         d S |�J ‚|�t!          ||¦  «         nt!          |||	z
  ¦  «         t!          |||
z
  ¦  «         d S )Nr…   r¥   r†   r-   r   ra   rL   r   rk   r   )r³   Úweights)rÄ   r³   )r   r   r   r   T©r0   rO   rÂ   )r2   r3   r4   r5   ÚabsÚaverageÚfinfor*   Úepsr   ÚsqrtÚsumr   r¬   r   rf   r   )rÂ   rF   rV   r<   r=   rW   r$   r@   rO   rµ   r¶   ÚX_sample_weight_avgÚX_sample_weight_varÚconstant_maskÚexpected_X_scaler·   r¸   r¹   rº   r»   r¼   s                        r'   Útest_preprocess_data_weightedrÐ   ©  s  € õ
 Œ)×
Ò
Ð 2Ñ
3Ô
3€CØ€IØ€Jð
 	�Š�˜JÑ'Ô'€AØ€A€aˆ#‚g�Jð €a€a€aˆ€d€G€G„Gˆr�M€G€G�Gð €A€a€a€aˆ€d�Gð €A€a€a€aˆ€d�GØ�Š�ÑÔ€Aå”F˜3Ÿ8š8 IÑ.Ô.Ñ/Ô/°!Ñ3€MÝ”j ¨°MÐBÑBÔB€OÝ”j ¨°MÐBÑBÔB€Oåœ* Q°ÀAÐFÑFÔFÐÝœ*Ø	
Ð Ñ	  QÑ&°ÀAðñ ô Ðð (¨"­r¬x¸¼Ñ/@Ô/@Ô/DÑ*DÒD€MÝ�} l l lÑ3Ô3Ð3Ý”wÐ2Ñ3Ô3µb´g¸m×>OÒ>OÑ>QÔ>QÑ6RÔ6RÑRÐð '(Ð�]Ñ#àÐ#ØÐ˜QÑÔˆå/?Ø	Ø	ØØ#Ø'ð0ñ 0ô 0Ñ,€BˆˆF�F˜G Wð Ð#à�IŠI‰KŒKˆØ�ZŠZ‰\Œ\ˆå˜f oÑ6Ô6Ð6Ý˜f oÑ6Ô6Ð6Ý˜g¥r¤w¨zÑ':Ô':Ñ;Ô;Ð;Øð 1Ý˜¥¤¨Ñ!7Ô!7Ñ8Ô8Ð8ØÐ'Ý˜B ¨¨¨¨4¨Ô 0°1Ñ 4Ñ5Ô5Ð5Ð5å˜B ¨¨¨¨4¨Ô 0°A¸Ñ4GÑ HÑIÔIÐIÝ˜˜G q¨?Ñ':Ñ;Ñ<Ô<Ð<Ð<Ð<àˆˆˆØÐ'Ý˜B Ñ"Ô"Ð"Ð"å˜B  OÑ 3Ñ4Ô4Ð4Ý˜˜A Ñ/Ñ0Ô0Ð0Ð0Ð0r)   Úlil_containerc                 óÊ  — t           j                             | ¦  «        }d}d}t          j        ||d|¬¦  «        } ||¦  «        }|                     |¦  «        }|                     ¦   «         }t          ||d¬¦  «        \  }}	}
}}}t          |
t          j        |¦  «        ¦  «         t          |d¦  «         t          |t          j	        |¦  «        ¦  «         |�J ‚t          |                     ¦   «         |¦  «         t          |	|¦  «         t          ||d¬¦  «        \  }}	}
}}}t          |
t          j
        |d¬	¦  «        ¦  «         t          |t          j
        |d¬	¦  «        ¦  «         t          |t          j	        |¦  «        ¦  «         |�J ‚t          |                     ¦   «         |¦  «         t          |	|t          j
        |d¬	¦  «        z
  ¦  «         d S )
Nr…   r   r-   )Údensityr�   Fr/   r   Tr²   )r2   r3   r4   r   r5   r¬   r   r   r´   rf   r¡   )rV   rÑ   r<   r=   rW   r$   r@   ÚXAr·   r¸   r¹   rº   r»   r¼   s                 r'   Ú#test_sparse_preprocess_data_offsetsrÕ   ô  sÓ  € å
Œ)×
Ò
Ð 2Ñ
3Ô
3€CØ€IØ€JÝŒ�I˜z°3ÀSÐIÑIÔI€AØˆ�aÑÔ€AØ�Š�ÑÔ€AØ	
�Š‰Œ€Bå/?Ø	ˆ1˜Eð0ñ 0ô 0Ñ,€BˆˆF�F˜G Wõ ˜f¥b¤h¨zÑ&:Ô&:Ñ;Ô;Ð;Ý˜f aÑ(Ô(Ð(Ý˜g¥r¤w¨zÑ':Ô':Ñ;Ô;Ð;Øˆ?ˆ?ˆ?Ý˜bŸjšj™lœl¨BÑ/Ô/Ð/Ý˜b !Ñ$Ô$Ð$å/?Ø	ˆ1˜Dð0ñ 0ô 0Ñ,€BˆˆF�F˜G Wõ ˜f¥b¤g¨b°qÐ&9Ñ&9Ô&9Ñ:Ô:Ð:Ý˜f¥b¤g¨a°aÐ&8Ñ&8Ô&8Ñ9Ô9Ð9Ý˜g¥r¤w¨zÑ':Ô':Ñ;Ô;Ð;Øˆ?ˆ?ˆ?Ý˜bŸjšj™lœl¨BÑ/Ô/Ð/Ý˜b !¥b¤g¨a°aÐ&8Ñ&8Ô&8Ñ"8Ñ9Ô9Ð9Ð9Ð9r)   c                 óœ   — t          ¦   «         \  }}d||dk     <    | |¦  «        }t          ||d¬¦  «        \  }}}}}}|j        dk    sJ ‚d S )Nr†   ç      @Tr/   Úcsr)r   r   Úformat)rƒ   r$   r@   rØ   Úcsr_rÀ   s         r'   Útest_csr_preprocess_datarÛ     si   € õ ÑÔ�D€A€qØ€A€aˆ#‚g�JØ
ˆ-˜Ñ
Ô
€CÝ*¨3°ÀÐFÑFÔFÑ€Dˆ!ˆQ��1�aØŒ;˜%ÒÐÐÐÐÐr)   Úto_copy)TFÚuse_sample_weight)FTc                 ó¦  — t          ¦   «         \  }}d||dk     <   |r!t          j        t          |¦  «        ¦  «        nd }| � | |¦  «        }t	          |||d|d¬¦  «        \  }}}}}}| �J|s|r#t          j        |j        |j        ¦  «        rJ ‚d S t          j        |j        |j        ¦  «        sJ ‚d S t          j        ||¦  «        | k    sJ ‚d S )Nr†   r×   TF)rO   r0   r«   Úcheck_input)r   r2   rf   Úlenr   Úmay_share_memoryÚdata)	rF   rÜ   rÝ   r$   r@   rO   ÚX_Úy_rÀ   s	            r'   Ú#test_preprocess_copy_data_no_checksrå     s	  € õ ÑÔ�D€A€qØ€A€aˆ#‚g�Jà'8ÐB•B”G�C ™FœF‘O”O�O¸d€MàÐ#ØÐ˜QÑÔˆå)Ø	Ø	Ø#ØØØðñ ô Ñ€BˆˆAˆq�!�Qð Ð#Øð 	8Ð'ð 	8åÔ*¨2¬7°A´FÑ;Ô;Ð;Ð;Ð;Ð;Ð;åÔ& r¤w°´Ñ7Ô7Ð7Ð7Ð7Ð7Ð7åÔ" 2 qÑ)Ô)°'¨kÒ:Ð:Ð:Ð:Ð:Ð:r)   c                 ó"	  — t           j                             |¦  «        }d}d}|                     ||¦  «        }|                     |¦  «        }|                     |¦  «        dz   }t          j        |t           j        ¬¦  «        }	t          j        |t           j        ¬¦  «        }
t          j        |t           j        ¬¦  «        }t          j        |t           j        ¬¦  «        }t          j        |t           j        ¬¦  «        }t          j        |t           j        ¬¦  «        }t          |	|
||| ¬¦  «        \  }}}}}}t          ||||| ¬¦  «        \  }}}}}}t          |	|||| ¬¦  «        \  }}}}}} t          ||
||| ¬¦  «        \  }!}"}#}$}%}&|j        t           j        k    sJ ‚|j        t           j        k    sJ ‚|j        t           j        k    sJ ‚|j        t           j        k    sJ ‚|j        t           j        k    sJ ‚| r|j        t           j        k    sJ ‚|j        t           j        k    sJ ‚|j        t           j        k    sJ ‚|j        t           j        k    sJ ‚|j        t           j        k    sJ ‚|j        t           j        k    sJ ‚| r|j        t           j        k    sJ ‚|j        t           j        k    sJ ‚|j        t           j        k    sJ ‚|j        t           j        k    sJ ‚|j        t           j        k    sJ ‚|j        t           j        k    sJ ‚| r| j        t           j        k    sJ ‚|!j        t           j        k    sJ ‚|"j        t           j        k    sJ ‚|#j        t           j        k    sJ ‚|$j        t           j        k    sJ ‚|j        t           j        k    sJ ‚| r|&j        t           j        k    sJ ‚|	j        t           j        k    sJ ‚|
j        t           j        k    sJ ‚|j        t           j        k    sJ ‚|j        t           j        k    sJ ‚t          ||dd¬¦  «         t          ||dd¬¦  «         t          ||d¬	¦  «         t          ||d¬	¦  «         t          ||¦  «         | rt          ||d¬	¦  «         d S d S )
Nr…   r   r   ©r*   rÅ   r    r   )ÚrtolÚatol©rè   )
r2   r3   r4   r5   ÚasarrayÚfloat32Úfloat64r   r*   r   )'rÂ   r0   rV   r<   r=   rW   r$   r@   ÚswÚX_32Úy_32Úsw_32ÚX_64Úy_64Úsw_64ÚXt_32Úyt_32Ú	X_mean_32Ú	y_mean_32Ú
X_scale_32Ú
sqrt_sw_32ÚXt_64Úyt_64Ú	X_mean_64Ú	y_mean_64Ú
X_scale_64Ú
sqrt_sw_64ÚXt_3264Úyt_3264ÚX_mean_3264Úy_mean_3264ÚX_scale_3264Úsqrt_sw_3264ÚXt_6432Úyt_6432ÚX_mean_6432Úy_mean_6432ÚX_scale_6432Úsqrt_sw_6432s'                                          r'   Útest_dtype_preprocess_datar  <  sÅ  € õ Œ)×
Ò
Ð 2Ñ
3Ô
3€CØ€IØ€JØ�Š�˜JÑ'Ô'€AØ�Š�ÑÔ€AØ	�Š�)Ñ	Ô	˜qÑ	 €BåŒ:�a�rœzÐ*Ñ*Ô*€DÝŒ:�a�rœzÐ*Ñ*Ô*€DÝŒJ�r¥¤Ð,Ñ,Ô,€EÝŒ:�a�rœzÐ*Ñ*Ô*€DÝŒ:�a�rœzÐ*Ñ*Ô*€DÝŒJ�r¥¤Ð,Ñ,Ô,€EåAQØØØ#ØØ'ðBñ Bô BÑ>€Eˆ5�)˜Y¨
°Jõ BRØØØ#ØØ'ðBñ Bô BÑ>€Eˆ5�)˜Y¨
°Jõ 	ØØØ'ØØ+ð	
ñ 	
ô 	
ñ K€GˆW�k ;°¸lõ 	ØØØ'ØØ+ð	
ñ 	
ô 	
ñ K€GˆW�k ;°¸lð Œ;�"œ*Ò$Ð$Ð$Ð$ØŒ;�"œ*Ò$Ð$Ð$Ð$ØŒ?�bœjÒ(Ð(Ð(Ð(ØŒ?�bœjÒ(Ð(Ð(Ð(ØÔ�rœzÒ)Ð)Ð)Ð)Øð .ØÔ¥2¤:Ò-Ð-Ð-Ð-àŒ;�"œ*Ò$Ð$Ð$Ð$ØŒ;�"œ*Ò$Ð$Ð$Ð$ØŒ?�bœjÒ(Ð(Ð(Ð(ØŒ?�bœjÒ(Ð(Ð(Ð(ØÔ�rœzÒ)Ð)Ð)Ð)Øð .ØÔ¥2¤:Ò-Ð-Ð-Ð-àŒ=�BœJÒ&Ð&Ð&Ð&ØŒ=�BœJÒ&Ð&Ð&Ð&ØÔ¥¤
Ò*Ð*Ð*Ð*ØÔ¥¤
Ò*Ð*Ð*Ð*ØÔ¥¤Ò+Ð+Ð+Ð+Øð 0ØÔ!¥R¤ZÒ/Ð/Ð/Ð/àŒ=�BœJÒ&Ð&Ð&Ð&ØŒ=�BœJÒ&Ð&Ð&Ð&ØÔ¥¤
Ò*Ð*Ð*Ð*ØÔ¥¤
Ò*Ð*Ð*Ð*ØÔ¥¤Ò+Ð+Ð+Ð+Øð 0ØÔ!¥R¤ZÒ/Ð/Ð/Ð/àŒ:�œÒ#Ð#Ð#Ð#ØŒ:�œÒ#Ð#Ð#Ð#ØŒ:�œÒ#Ð#Ð#Ð#ØŒ:�œÒ#Ð#Ð#Ð#å�E˜5 t°$Ð7Ñ7Ô7Ð7Ý�E˜5 t°$Ð7Ñ7Ô7Ð7Ý�I˜y¨tÐ4Ñ4Ô4Ð4Ý�I˜y¨tÐ4Ñ4Ô4Ð4Ý�J 
Ñ+Ô+Ð+Øð ;Ý˜
 J°TÐ:Ñ:Ô:Ð:Ð:Ð:ð;ð ;r)   c                  óæ  — t          ddd¬¦  «        \  } }t          j        ddgddgd	d
ggt          j        ¬¦  «        }t          j        g d¢t          j        ¬¦  «        }|                      ||¬¦  «        }|                      ||¬¦  «        }t	          d¬¦  «        5  t          ||dd¬¦  «        ^}}}d d d ¦  «         n# 1 swxY w Y   |j        | j        k    sJ ‚|j        | j        k    sJ ‚d S )NÚtorchÚmpsrì   )Údevice_nameÚ
dtype_namer   r   rk   r¥   rI   rH   rç   )r   r   r¥   )ÚdeviceT©Úarray_api_dispatch)r0   rß   )r   r2   rë   Úint64r   r   r*   rì   )	Úxpr  ÚX_npÚy_npÚX_xpÚy_xpÚX_outÚy_outrÀ   s	            r'   Ú=test_preprocess_data_integer_array_api_on_float32_only_devicer  Ÿ  sE  € Ý% g¸5ÈYÐWÑWÔW�J€Bˆõ Œ:˜˜1�v  1˜v¨¨1 vÐ.µb´hÐ?Ñ?Ô?€DÝŒ:�i�i�i¥r¤xÐ0Ñ0Ô0€DØ�:Š:�d 6ˆ:Ñ*Ô*€DØ�:Š:�d 6ˆ:Ñ*Ô*€Då	¨4Ð	0Ñ	0Ô	0ð 
ð 
Ý+Ø�$ d¸ð
ñ 
ô 
Ðˆˆu�qð
ð 
ð 
ñ 
ô 
ð 
ð 
ð 
ð 
ð 
ð 
øøøð 
ð 
ð 
ð 
ð
 Œ;˜"œ*Ò$Ð$Ð$Ð$ØŒ;˜"œ*Ò$Ð$Ð$Ð$Ð$Ð$s   ÂCÃCÃ	CÚ	n_targetsr   c                 ó  — t           j                             |¦  «        }d}d}d|                     |¦  «        z   }|                     ||¦  «        }| €|                     |¦  «        }n|                     || ¦  «        }t          j        |¦  «        }	||	d d …t           j        f         z  }
| €||	z  }n||	d d …t           j        f         z  }|�8 ||¦  «        }| €  ||                     dd¦  «        ¦  «        }n ||¦  «        }t          |||¦  «        \  }}}t          ||	¦  «         |�>| 	                    ¦   «         }| 	                    ¦   «         }| €| 
                    ¦   «         }t          ||
¦  «         t          ||¦  «         d S )Nr…   r   rL   éÿÿÿÿr   )r2   r3   r4   r5   rÊ   ÚnewaxisÚreshaper   r   r¬   r~   )r  rF   rV   r<   r=   rW   rO   r$   r@   Úexpected_sqrt_swÚexpected_rescaled_XÚexpected_rescaled_yÚ
rescaled_XÚ
rescaled_yr¼   s                  r'   Útest_rescale_datar)  ²  s°  € õ Œ)×
Ò
Ð 2Ñ
3Ô
3€CØ€IØ€Jà˜#Ÿ(š( 9Ñ-Ô-Ñ-€MØ�Š�˜JÑ'Ô'€AØÐØ�HŠH�YÑÔˆˆà�HŠH�Y 	Ñ*Ô*ˆå”w˜}Ñ-Ô-ÐØÐ.¨q¨q¨qµ"´*¨}Ô=Ñ=ÐàÐØÐ"2Ñ2ÐÐàÐ"2°1°1°1µb´j°=Ô"AÑAÐàÐ#ØÐ˜QÑÔˆØÐØ Ð  §¢¨2¨qÑ!1Ô!1Ñ2Ô2ˆAˆAà Ð  Ñ#Ô#ˆAå&3°A°q¸-Ñ&HÔ&HÑ#€J�
˜Gå�GÐ-Ñ.Ô.Ð.àÐ#Ø×'Ò'Ñ)Ô)ˆ
Ø×'Ò'Ñ)Ô)ˆ
ØÐØ#×)Ò)Ñ+Ô+ˆJå�JÐ 3Ñ4Ô4Ð4Ý�JÐ 3Ñ4Ô4Ð4Ð4Ð4r)   c                 ó\  — t          ¦   «         }|j                             t          j        ¦  «        }|j                             t          j        ¦  «        } | |¦  «        }t          j        |j        t          j        ¬¦  «        }|j                             t          j        ¦  «        }|j                             t          j        ¦  «        } | |¦  «        }t          j        |j        t          j        ¬¦  «        }	t          |||¦  «        \  }
}t          |||	¦  «        \  }}|
 
                    ¦   «         \  }}}}| 
                    ¦   «         \  }}}}|\  }}}|\  }}}|j        t          j        k    sJ ‚|j        t          j        k    sJ ‚t          ||t          ¬¦  «         t          |||¦  «        \  }}t          |||	¦  «        \  }}| 
                    ¦   «         \  }}}}| 
                    ¦   «         \  }}}}|\  }}}|\  }}}|j        t          j        k    sJ ‚|j        t          j        k    sJ ‚t          ||t          ¬¦  «         t          ||t          ¬¦  «         t          ||¦  «         t          ||¦  «         t          ||¦  «         t          ||¦  «         d S )Nrç   rê   )r   râ   r7   r2   rì   ÚtargetÚarangerK   rí   r   Ú_next_pyr*   r   rè   r   )rƒ   Úirisrï   rð   ÚX_csr_32Úsample_weight_32rò   ró   ÚX_csr_64Úsample_weight_64Ú
dataset_32rÀ   Ú
dataset_64Úxi_32Úyi_32Úxi_64Úyi_64Ú
xi_data_32Ú
xi_data_64Údatasetcsr_32Údatasetcsr_64Úxicsr_32Úyicsr_32Úxicsr_64Úyicsr_64Úxicsr_data_32Úxicsr_data_64s                              r'   Útest_fused_types_make_datasetrC  Ý  sŠ  € å‰;Œ;€DàŒ9×Ò�BœJÑ'Ô'€DØŒ;×Ò�bœjÑ)Ô)€DØˆ}˜TÑ"Ô"€HÝ”y ¤µ"´*Ð=Ñ=Ô=ÐàŒ9×Ò�BœJÑ'Ô'€DØŒ;×Ò�bœjÑ)Ô)€DØˆ}˜TÑ"Ô"€HÝ”y ¤µ"´*Ð=Ñ=Ô=Ðõ !  tÐ-=Ñ>Ô>�M€J�Ý   tÐ-=Ñ>Ô>�M€J�Ø#×,Ò,Ñ.Ô.Ñ€Eˆ5�!�QØ#×,Ò,Ñ.Ô.Ñ€Eˆ5�!�QØÑ€J��1ØÑ€J��1àÔ�rœzÒ)Ð)Ð)Ð)ØÔ�rœzÒ)Ð)Ð)Ð)Ý�E˜5¥tÐ,Ñ,Ô,Ð,õ $ H¨dÐ4DÑEÔEÑ€M�1Ý# H¨dÐ4DÑEÔEÑ€M�1Ø,×5Ò5Ñ7Ô7Ñ€Hˆh˜˜1Ø,×5Ò5Ñ7Ô7Ñ€Hˆh˜˜1Ø"Ñ€M�1�aØ"Ñ€M�1�aàÔ¥"¤*Ò,Ð,Ð,Ð,ØÔ¥"¤*Ò,Ð,Ð,Ð,å�M =µtÐ<Ñ<Ô<Ð<Ý�H˜h­TÐ2Ñ2Ô2Ð2å�z =Ñ1Ô1Ð1Ý�z =Ñ1Ô1Ð1Ý�u˜hÑ'Ô'Ð'Ý�u˜hÑ'Ô'Ð'Ð'Ð'r)   ÚX_shape)©ra   rI   )ra   é   )r}   r}   c                 óv   — g | ]6}t          j        |t           j                             d ¬¦  «        ¬¦  «        ‘Œ7S )z/Known to fail for CSR arrays, see issue #30131.)Úreason)Úmarks)r:   ÚparamÚmarkÚxfail)Ú.0Ú	containers     r'   ú
<listcomp>rO    s[   € ð ð ð ð õ 	ŒØÝ”+×#Ò#ØHð $ñ ô ð	
ñ 	
ô 	
ðð ð r)   c                 ó  — t           j                             |¦  «        }| \  }}|                     ||¦  «        }|                     |¦  «        }|� ||¦  «        }t	          |¬¦  «        }	t          di |	¤Ž                     ||d¬¦  «        }
|
j                             ¦   «         }|r|
j	        }t          j
        |¦  «        }|
                     |||¬¦  «         t          |
j        |d¬¦  «         |rt          |
j	        |¦  «         d}|
                     |||¬¦  «         t          |
j        |d¬¦  «         |rt          |
j	        |¦  «         |                     dd|j        d	         ¬
¦  «        }|
                     |||¬¦  «        }
|
j                             ¦   «         }|r|
j	        }|
                     ||t           j        |z  ¬¦  «         t          |
j        ||€dnd¬¦  «         |rt          |
j	        |¦  «         |                     ¦   «         }d	|dd…<   |dd…xx         dz  cc<   |
                     |||¬¦  «         |
j                             ¦   «         }|r|
j	        }|
                     |dd…         |dd…         |dd…         ¬¦  «         t          |
j        |d¬¦  «         |rt          |
j	        |¦  «         |�$t          j        ||d|dz  …         gd¬¦  «        }n#t          j        ||d|dz  …         gd	¬¦  «        }t          j        ||d|dz  …         g¦  «        }|                     ¦   «         }|d|dz  …xx         dz  cc<   t          j        ||d|dz  …         gd	¬¦  «        }t          di |	¤Ž                     |||¬¦  «        }t          di |	¤Ž                     |||¬¦  «        }t          |j        |j        d¬¦  «         |rt          |j	        |j	        ¦  «         dS dS )zõTest that the impact of sample_weight is consistent.

    Note that this test is stricter than the common test
    check_sample_weight_equivalence alone and also tests sparse X.
    It is very similar to test_enet_sample_weight_consistency.
    Nr/   rN   r   rê   g     À^@g{®Gáz„?r   r   )ÚlowÚhighrK   gñhãˆµøä>éûÿÿÿiè  Úcsc)rÙ   r²   r‡   )r2   r3   r4   r5   rˆ   r
   r    r!   r«   r"   Ú	ones_liker   rQ   rR   Úpir   r“   Úconcatenate)rD  rF   r0   rV   r<   r=   rW   r$   r@   rŠ   r&   ÚcoefÚ	interceptrO   Úsample_weight_0Úcoef_0Úintercept_0ru   Úy2Úsample_weight_1Úsample_weight_2Úreg1Úreg2s                          r'   Ú0test_linear_regression_sample_weight_consistencyrb    s]  € õ0 Œ)×
Ò
Ð 2Ñ
3Ô
3€CØ#Ñ€Iˆzà�Š�˜JÑ'Ô'€AØ�Š�ÑÔ€AØÐ#ØÐ˜QÑÔˆÝ Ð.Ñ.Ô.€Få
Ð
$Ð
$˜VÐ
$Ð
$×
(Ò
(¨¨A¸TÐ
(Ñ
BÔ
B€CØŒ9�>Š>ÑÔ€DØð #Ø”Nˆ	õ
 ”L ‘O”O€MØ‡G‚GˆAˆq €GÑ.Ô.Ð.Ý�C”I˜t¨$Ð/Ñ/Ô/Ð/Øð 3Ý˜œ¨	Ñ2Ô2Ð2ð €MØ‡G‚GˆAˆq €GÑ.Ô.Ð.Ý�C”I˜t¨$Ð/Ñ/Ô/Ð/Øð 3Ý˜œ¨	Ñ2Ô2Ð2ð —K’K D¨q°q´w¸q´z�KÑBÔB€MØ
�'Š'�!�Q mˆ'Ñ
4Ô
4€CØŒ9�>Š>ÑÔ€DØð #Ø”Nˆ	à‡G‚GˆAˆq¥¤¨Ñ 5€GÑ6Ô6Ð6Ý�C”I˜tÐ2BÐ2J¨$¨$ÐPTÐUÑUÔUÐUØð 3Ý˜œ¨	Ñ2Ô2Ð2ð $×(Ò(Ñ*Ô*€OØ€O�B�C�CÑØ€b€c€c€F€F„Fˆd�N€F€F�FØ‡G‚GˆAˆq €GÑ0Ô0Ð0ØŒY�^Š^ÑÔ€FØð %Ø”nˆØ‡G‚GˆAˆcˆrˆcŒF�A�c�r�c”F¨-¸¸¸Ô*<€GÑ=Ô=Ð=Ý�C”I˜v¨DÐ1Ñ1Ô1Ð1Øð 5Ý˜œ¨Ñ4Ô4Ð4ð Ð#ÝŒ]˜A˜qÐ!1 9°¡>Ð!1Ô2Ð3¸EÐBÑBÔBˆˆåŒ^˜Q Ð"2 I°¡NÐ"2Ô 3Ð4¸1Ð=Ñ=Ô=ˆÝ	Œ˜˜AÐ. 	¨Q¡Ð.Ô/Ð0Ñ	1Ô	1€BØ#×(Ò(Ñ*Ô*€OØÐ$�i 1‘nÐ$Ð%Ð%Ô%¨Ñ*Ð%Ð%Ñ%Ý”nØ	˜Ð&6¨	°Q©Ð&6Ô7Ð8¸qðñ ô €Oõ Ð%Ð%˜fÐ%Ð%×)Ò)¨!¨Q¸oÐ)ÑNÔN€DÝÐ%Ð%˜fÐ%Ð%×)Ò)¨"¨bÀÐ)ÑPÔP€DÝ�D”J ¤
°Ð6Ñ6Ô6Ð6Øð :Ý˜œ¨¬Ñ9Ô9Ð9Ð9Ð9ð:ð :r)   c                  ó°  — t          j        d¦  «        } t          j                             d¦  «        }|                     d¬¦  «        }|                     d¬¦  «        }t          ¦   «                              ||¦  «        }|                      |¦  «        }| 	                    |¦  «         t          d¬¦  «        5  t          j        t          d¬	¦  «        5  | 	                    |¦  «         d d d ¦  «         n# 1 swxY w Y   t          |¦  «        \  }}}t          |||¦  «        }| 	                    |¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )
NÚarray_api_strictr   rE  rJ   ra   Tr  z.*must use the same namespacerd   )r:   Úimportorskipr2   r3   Údefault_rngrP   r
   r    rë   r#   r   rg   Ú
ValueErrorr   r   )	r  r<   r$   r@   r&   r  Ú	xp_targetrÀ   r  s	            r'   Ú test_array_api_move_estimator_tori  n  s³  € å	Ô	Ð/Ñ	0Ô	0€BÝ
Œ)×
Ò
 Ñ
"Ô
"€CØ�
Š
˜ˆ
Ñ Ô €AØ�
Š
˜ˆ
ÑÔ€Aå
Ñ
Ô
×
 Ò
   AÑ
&Ô
&€CØ�:Š:�a‰=Œ=€DØ‡K‚K�ÑÔÐå	¨4Ð	0Ñ	0Ô	0ð ð ÝŒ]�:Ð-LÐMÑMÔMð 	ð 	Ø�KŠK˜ÑÔÐð	ð 	ð 	ñ 	ô 	ð 	ð 	ð 	ð 	ð 	ð 	øøøð 	ð 	ð 	ð 	å7¸Ñ=Ô=Ñˆ	�1�fÝ  Y°Ñ7Ô7ˆØ�Š�DÑÔÐðð ð ñ ô ð ð ð ð ð ð ð øøøð ð ð ð ð ð s6   Â<EÃC:Ã.EÃ:C>	Ã>EÄC>	Ä<EÅEÅEc                  ó  —  G d„ dt           t          ¦  «        }  | ¦   «         }|                     dd¬¦  «         |                     t	          j        d¦  «        ¦  «        }t          t	          j        |d¬¦  «        d¦  «         dS )z~Test that _predict_proba_lr of LinearClassifierMixin deals with large
    negative values.

    Note that exp(-1000) = 0.
    c                   ó    — e Zd Zd„ Zd„ Zd„ ZdS )ú:test_predict_proba_lr_large_values.<locals>.MockClassifierc                 ó   — d S )Nr‡   )Úselfs    r'   Ú__init__zCtest_predict_proba_lr_large_values.<locals>.MockClassifier.__init__‰  s   € ØˆDr)   c                 ó   — d| _         d S )NT)Ú__sklearn_is_fitted__)rn  r$   r@   s      r'   r    z>test_predict_proba_lr_large_values.<locals>.MockClassifier.fitŒ  s   € Ø)-ˆDÔ&Ð&Ð&r)   c                 óR   — |j         d         }t          j        dgdz  |dg¦  «        S )Nr   g     @�Àr¥   r   )rR   r2   Útile)rn  r$   r=   s      r'   Údecision_functionzLtest_predict_proba_lr_large_values.<locals>.MockClassifier.decision_function�  s)   € Øœ œ
ˆIÝ”7˜G˜9 q™=¨9°a¨.Ñ9Ô9Ð9r)   N)Ú__name__Ú
__module__Ú__qualname__ro  r    rt  r‡   r)   r'   ÚMockClassifierrl  ˆ  sA   € € € € € ð	ð 	ð 	ð	.ð 	.ð 	.ð	:ð 	:ð 	:ð 	:ð 	:r)   rx  N)r$   r@   rI   r   r²   )r   r   r    Ú_predict_proba_lrr2   rf   r   rË   )rx  ÚclfÚprobas      r'   Ú"test_predict_proba_lr_large_valuesr|  �  s”   € ð	:ð 	:ð 	:ð 	:ð 	:Õ.µñ 	:ô 	:ð 	:ð ˆ.Ñ
Ô
€CØ‡G‚Gˆd�d€GÑÔÐà×!Ò!¥"¤'¨!¡*¤*Ñ-Ô-€EÝ•B”F˜5 qÐ)Ñ)Ô)¨1Ñ-Ô-Ð-Ð-Ð-r)   )GÚnumpyr2   r:   Úscipyr   r   Úsklearnr   Úsklearn.baser   Úsklearn.datasetsr   r   r	   Úsklearn.linear_modelr
   Úsklearn.linear_model._baser   r   r   r   Úsklearn.preprocessingr   Úsklearn.utils._array_apir   r   Úsklearn.utils._testingr   r   r   r   r   Úsklearn.utils.fixesr   r   r   r   r   rè   r(   rK  Úparametrizerí   rì   rE   r_   rj   rq   r{   r‚   r�   r–   r™   r›   rž   r£   r¦   r°   r½   rÁ   rÐ   rÕ   rÛ   rå   r  r  r)  rC  rb  ri  r|  r‡   r)   r'   ú<module>r‰     sê  ðð Ð Ð Ð Ø €€€Ø  Ð  Ð  Ð  Ð  Ð  Ð  Ð  à "Ð "Ð "Ð "Ð "Ð "Ø &Ð &Ð &Ð &Ð &Ð &Ø QÐ QÐ QÐ QÐ QÐ QÐ QÐ QÐ QÐ QØ 1Ð 1Ð 1Ð 1Ð 1Ð 1ðð ð ð ð ð ð ð ð ð ð ð ð 4Ð 3Ð 3Ð 3Ð 3Ð 3Ø PÐ PÐ PÐ PÐ PÐ PÐ PÐ Pðð ð ð ð ð ð ð ð ð ð ð ð ð ðð ð ð ð ð ð ð ð ð ð ð ð ð ð €ð3ð 3ð 3ð0 „×Ò˜ 2¤:¨r¬zÐ":Ñ;Ô;ð?ð ?ñ <Ô;ð?ð0 „×ÒÐ+¨d¨V°nÑ-DÑEÔEØ„×Ò˜¨4°¨-Ñ8Ô8ð$+ð $+ñ 9Ô8ñ FÔEð$+ðN	ð 	ð 	ð „×ÒÐ0°6¸6Ð2BÑCÔCð'ð 'ñ DÔCð'ð"Pð Pð Pð&=ð =ð =ð „×Ò˜¨4°¨-Ñ8Ô8Ø„×Ò˜¨.Ñ9Ô9ð7ð 7ñ :Ô9ñ 9Ô8ð7ð$Pð Pð Pð" „×Ò˜¨.Ñ9Ô9ðPð Pñ :Ô9ðPð")ð )ð )ð.;ð ;ð ;ð"
9ð 
9ð 
9ð9ð 9ð 9ð" „×ÒÐ+¨d¨V°nÑ-DÑEÔEØ„×Ò˜ D¨% =Ñ1Ô1ð39ð 39ñ 2Ô1ñ FÔEð39ðl7ð 7ð 7ð< „×ÒÐ+¨d¨V°nÑ-DÑEÔEð.ð .ñ FÔEð.ð* „×ÒÐ*¨U°D¨MÑ:Ô:Ø„×ÒÐ+¨d¨V°nÑ-DÑEÔEðF1ð F1ñ FÔEñ ;Ô:ðF1ðR „×Ò˜¨.Ñ9Ô9ð:ð :ñ :Ô9ð:ð< „×Ò˜¨.Ñ9Ô9ð ð  ñ :Ô9ð ð „×ÒÐ+¨d¨V°nÑ-DÑEÔEØ„×Ò˜ MÑ2Ô2Ø„×ÒÐ,¨mÑ<Ô<ð;ð ;ñ =Ô<ñ 3Ô2ñ FÔEð;ð8 „×ÒÐ*¨U°D¨MÑ:Ô:Ø„×Ò˜¨5°$¨-Ñ8Ô8ð^;ð ^;ñ 9Ô8ñ ;Ô:ð^;ðB%ð %ð %ð& „×Ò˜ t¨Q iÑ0Ô0Ø„×ÒÐ+¨d¨V°nÑ-DÑEÔEð&5ð &5ñ FÔEñ 1Ô0ð&5ðR „×Ò˜¨.Ñ9Ô9ð*(ð *(ñ :Ô9ð*(ðZ „×Ò˜Ð$CÐ$CÐ$CÑDÔDØ„×ÒØØ	€Fðð ð (ðñ ô ñ	ñô ð „×Ò˜¨5°$¨-Ñ8Ô8ðQ:ð Q:ñ 9Ô8ñô ñ EÔDðQ:ðh )ðð ñ )Ô(ðð$.ð .ð .ð .ð .r)   